From 9168b38e565765d9069cfdeddd62a42b2bd4eb18 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Wed, 24 May 2023 13:28:20 +0000
Subject: [PATCH] Remove unused hardware, software and date
---
dan/manager/training.py | 18 ------------------
1 file changed, 18 deletions(-)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index 5821d2e8..cb0673f0 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)
--
GitLab