Sha256: e44ecac76e7bc1374fb6c415e963255270c54789c8017048d3fa7bdf7fa7058f
Contents?: true
Size: 629 Bytes
Versions: 54
Compression:
Stored size: 629 Bytes
Contents
module Torch
module NN
class InstanceNorm < BatchNorm
def initialize(num_features, eps: 1e-5, momentum: 0.1, affine: false, track_running_stats: false)
super(num_features, eps: eps, momentum: momentum, affine: affine, track_running_stats: track_running_stats)
end
def forward(input)
_check_input_dim(input)
F.instance_norm(
input, running_mean: @running_mean, running_var: @running_var,
weight: @weight, bias: @bias,
use_input_stats: @training || !@track_running_stats,
momentum: @momentum, eps: @eps
)
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems