def finish(self, resources: Resources): # pylint: disable=unused-argument
"""
Releasing resources and saving models.
Args:
resources: The resources used by the training process.
Returns:
"""
if self.resource:
keys_to_serializers = {}
for key in resources.keys():
# pylint: disable=consider-using-with
if key == "model":
keys_to_serializers[key] = lambda x, y: pickle.dump(
x.state_dict(), open(y, "wb"))
else:
keys_to_serializers[key] = lambda x, y: pickle.dump(
x, open(y, "wb"))
self.resource.save(keys_to_serializers,
output_dir=self.config_model.resource_dir)
self.__save_model_checkpoint()
def finish(self, resources: Resources): # pylint: disable=unused-argument
if self.resource:
keys_to_serializers = {}
for key in resources.keys():
if key == "model":
keys_to_serializers[key] = \
lambda x, y: pickle.dump(x.state_dict(), open(y, "wb"))
else:
keys_to_serializers[key] = \
lambda x, y: pickle.dump(x, open(y, "wb"))
self.resource.save(keys_to_serializers)
self.save_model_checkpoint()