Contents

module Torch
  module Utils
    module Data
      class DataLoader
        def initialize(dataset, batch_size: 1)
          @dataset = dataset
          @batch_size = batch_size
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
torch-rb-0.1.1	lib/torch/utils/data/data_loader.rb
torch-rb-0.1.0	lib/torch/utils/data/data_loader.rb