From 68e91e25bd70c39397063747724597d056bd0699 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Mon, 24 Jun 2024 15:09:40 +0200
Subject: [PATCH] fix para
---
main_ray_tune.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/main_ray_tune.py b/main_ray_tune.py
index 0ff6221..f4cd748 100644
--- a/main_ray_tune.py
+++ b/main_ray_tune.py
@@ -199,9 +199,9 @@ def test_best_model(best_result, args):
if torch.cuda.is_available():
device = "cuda:0"
if torch.cuda.device_count() > 1:
- print(print(type(best_trained_model.module)))
+ print(type(best_trained_model))
best_trained_model = torch.nn.DataParallel(best_trained_model)
- print(print(type(best_trained_model.module)))
+ print(type(best_trained_model))
best_trained_model.to(device)
criterion_rt = torch.nn.MSELoss()
--
GitLab