Sha256: 80d27a9ae393393f37a3c9e7642ad22f204528522307e5e7eeac138aa6bd0db8
Contents?: true
Size: 856 Bytes
Versions: 59
Compression:
Stored size: 856 Bytes
Contents
module Torch
module Optim
module LRScheduler
class LRScheduler
def initialize(optimizer, last_epoch)
@optimizer = optimizer
if last_epoch == -1
optimizer.param_groups.each do |group|
group[:initial_lr] ||= group[:lr]
end
last_epoch = 0
else
raise NotImplementedYet
end
@base_lrs = optimizer.param_groups.map { |group| group[:initial_lr] }
@last_epoch = last_epoch
@step_count = 0
step(last_epoch)
end
def step(epoch = nil)
@step_count += 1
epoch ||= @last_epoch + 1
@last_epoch = epoch
@optimizer.param_groups.zip(get_lr).each do |param_group, lr|
param_group[:lr] = lr
end
end
end
end
end
end
Version data entries
59 entries across 59 versions & 1 rubygems