(self, optimizer, scheduler=None)
| 77 | self.ulb_dset = dset |
| 78 | |
| 79 | def set_optimizer(self, optimizer, scheduler=None): |
| 80 | self.optimizer = optimizer |
| 81 | self.scheduler = scheduler |
| 82 | |
| 83 | def warmup(self, args, logger=None): |
| 84 | ngpus_per_node = torch.cuda.device_count() |
nothing calls this directly
no outgoing calls
no test coverage detected