def save(self, ckpt=None):
'''Save net models for algorithm given the required property self.net_names'''
if not hasattr(self, 'net_names'):
logger.info(
'No net declared in self.net_names in init_nets(); no models to save.'
)
else:
net_util.save_algorithm(self, ckpt=ckpt)
def save(self, epi=None):
'''Save net models for algorithm given the required property self.net_names'''
if not hasattr(self, 'net_names'):
logger.info('No net declared in self.net_names in init_nets(); no models to save.')
else:
net_util.save_algorithm(self, epi=epi)