MCPcopy Create free account
hub / github.com/TorchSSL/TorchSSL / set_data_loader

Method set_data_loader

models/pimodel/pimodel.py:59–61  ·  view source on GitHub ↗
(self, loader_dict)

Source from the content-addressed store, hash-verified

57 self.bn_controller = Bn_Controller()
58
59 def set_data_loader(self, loader_dict):
60 self.loader_dict = loader_dict
61 self.print_fn(f'[!] data loader keys: {self.loader_dict.keys()}')
62
63 def set_optimizer(self, optimizer, scheduler=None):
64 self.optimizer = optimizer

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected