Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def reset_child_params(module):
- for layer in module.children():
- if hasattr(layer, "reset_parameters"):
- layer.reset_parameters()
- reset_child_params(layer)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement