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"