Module: Torchrb::NN::TrainerDefault

Defined in:
lib/torchrb/nn/trainer_default.rb

Instance Method Summary collapse

Instance Method Details

#define_trainer(options) ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
15
 
# File 'lib/torchrb/nn/trainer_default.rb', line 3

def define_trainer options
  torch.eval "    number_of_iterations = \#{options.fetch(:iterations){50}} -- Must be set for the callback to work\n\n    criterion = nn.ClassNLLCriterion()\n\n    trainer = nn.StochasticGradient(net, criterion)\n    trainer.verbose = false\n    trainer.learningRate = \#{options.fetch(:learning_rate){0.001}}\n    trainer.maxIteration = number_of_iterations\n    trainer.hookIteration = iteration_callback\n  EOF\nend\n", __FILE__, __LINE__