A module that wraps another layer in which some weights will be replaced by 0 during training.

WeightDropout(module, weight_p, layer_names = "weight_hh_l0")

Arguments

module

module

weight_p

weight_p

layer_names

layer_names

Value

None