Get bare model, especially under wrapping with DistributedDataParallel or DataParallel.
(self, net)
| 137 | raise NotImplementedError(f'Scheduler {scheduler_type} is not implemented yet.') |
| 138 | |
| 139 | def get_bare_model(self, net): |
| 140 | """Get bare model, especially under wrapping with |
| 141 | DistributedDataParallel or DataParallel. |
| 142 | """ |
| 143 | if isinstance(net, (DataParallel, DistributedDataParallel)): |
| 144 | net = net.module |
| 145 | return net |
| 146 | |
| 147 | @master_only |
| 148 | def print_network(self, net): |
no outgoing calls
no test coverage detected