Exemplo n.º 1
0
 def train_outputs(self):
     return datasets_registry.num_classes(self.dataset_hparams)
Exemplo n.º 2
0
 def pretrain_outputs(self):
     datasets_registry.num_classes(self.pretrain_dataset_hparams)