Remove unused hardware, software and date

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)