diff --git a/dan/manager/training.py b/dan/manager/training.py index 5821d2e804f3129a263f75da0723c3c5888c97af..cb0673f0e635ff2fd8c64907eafb45a90f7a9173 100644 --- a/dan/manager/training.py +++ b/dan/manager/training.py @@ -4,8 +4,6 @@ import json import os import pickle import random -import sys -from datetime import date from time import time import numpy as np @@ -523,7 +521,6 @@ class GenericTrainingManager: return params = copy.deepcopy(self.params) params = class_to_str_dict(params) - params["date"] = date.today().strftime("%d/%m/%Y") total_params = 0 for model_name in self.models.keys(): current_params = compute_nb_params(self.models[model_name]) @@ -533,21 +530,6 @@ class GenericTrainingManager: ] total_params += current_params params["model_params"]["total_params"] = "{:,}".format(total_params) - - params["hardware"] = dict() - if self.device != "cpu": - for i in range(self.params["training_params"]["nb_gpu"]): - params["hardware"][str(i)] = "{} {}".format( - torch.cuda.get_device_name(i), torch.cuda.get_device_properties(i) - ) - else: - params["hardware"]["0"] = "CPU" - params["software"] = { - "python_version": sys.version, - "pytorch_version": torch.__version__, - "cuda_version": torch.version.cuda, - "cudnn_version": torch.backends.cudnn.version(), - } with open(path, "w") as f: json.dump(params, f, indent=4)