lib/torch.rb in torch-rb-0.2.3 vs lib/torch.rb in torch-rb-0.2.4
- old
+ new
@@ -172,9 +172,10 @@
require "torch/nn/functional"
require "torch/nn/init"
# utils
require "torch/utils/data/data_loader"
+require "torch/utils/data/dataset"
require "torch/utils/data/tensor_dataset"
# hub
require "torch/hub"