Scale the parameters of a module and return it.
(module, scale)
| 31 | return module |
| 32 | |
| 33 | def scale_module(module, scale): |
| 34 | """ |
| 35 | Scale the parameters of a module and return it. |
| 36 | """ |
| 37 | for p in module.parameters(): |
| 38 | p.detach().mul_(scale) |
| 39 | return module |
| 40 | |
| 41 | |
| 42 | def conv_nd(dims, *args, **kwargs): |
nothing calls this directly
no outgoing calls
no test coverage detected