You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
1234567 |
- import torch
-
- def mutate_remove_dropout(model):
- for module in model.modules():
- if isinstance(module, torch.nn.Dropout):
- module._backup_p = module.p
- module.p = 0
|