From 9c84b875a36d6ef62b0f9b380078448db7ec5a58 Mon Sep 17 00:00:00 2001 From: Victor Mylle Date: Sun, 26 Nov 2023 22:27:35 +0000 Subject: [PATCH] Changing some things --- src/trainers/trainer.py | 23 ++++++++++++++--------- src/utils/clearml.py | 1 + 2 files changed, 15 insertions(+), 9 deletions(-) diff --git a/src/trainers/trainer.py b/src/trainers/trainer.py index dc388cf..cc8e912 100644 --- a/src/trainers/trainer.py +++ b/src/trainers/trainer.py @@ -4,6 +4,7 @@ from src.utils.clearml import ClearMLHelper import plotly.graph_objects as go import numpy as np from plotly.subplots import make_subplots +from clearml.config import running_remotely class Trainer: @@ -51,19 +52,23 @@ class Trainer: if not self.clearml_helper: return None - task_name = input("Enter a task name: ") - if task_name == "": - task_name = "Untitled Task" - task = self.clearml_helper.get_task(task_name=task_name) + + task = self.clearml_helper.get_task(task_name="None") + + # check if running remotely + + if not running_remotely(): + task_name = input("Enter a task name: ") + task.set_name(task_name) + + # change_description = input("Enter a change description: ") + change_description = "" + if change_description: + task.set_comment(change_description) if self.debug: task.add_tags("Debug") - # change_description = input("Enter a change description: ") - change_description = "" - if change_description: - task.set_comment(change_description) - task.add_tags(self.model.__class__.__name__) task.add_tags(self.criterion.__class__.__name__) task.add_tags(self.optimizer.__class__.__name__) diff --git a/src/utils/clearml.py b/src/utils/clearml.py index af2c623..42cb997 100644 --- a/src/utils/clearml.py +++ b/src/utils/clearml.py @@ -5,6 +5,7 @@ class ClearMLHelper: self.project_name = project_name def get_task(self, task_name: str = "Model Training"): + Task.add_requirements("../../requirements.txt") Task.ignore_requirements("torch") Task.ignore_requirements("torchvision") Task.ignore_requirements("tensorboard")