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

Version Path
torch-rb-0.19.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.19.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.18.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.17.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.17.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.16.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.15.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.14.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.14.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.13.2 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.13.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.13.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.12.2 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.12.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.12.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.11.2 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.11.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.11.0 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.10.2 lib/torch/optim/lr_scheduler/lr_scheduler.rb
torch-rb-0.10.1 lib/torch/optim/lr_scheduler/lr_scheduler.rb