VictorMylle/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added time as input feature

Browse Source
This commit is contained in:
Victor Mylle
2023-11-26 18:43:03 +00:00
parent a2c9427d16
commit 2f40f41786
9 changed files with 168 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/trainers/trainer.py
View File

@@ -41,7 +41,7 @@ class Trainer:
self.patience = patience
self.delta = delta
def add_metrics_to_track(self, loss: torch.nn.Module | list[torch.nn.Module]):
def add_metrics_to_track(self, loss):
if isinstance(loss, list):
self.metrics_to_track.extend(loss)
else:
@@ -59,7 +59,8 @@ class Trainer:
if self.debug:
task.add_tags("Debug")
change_description = input("Enter a change description: ")
# change_description = input("Enter a change description: ")
change_description = ""
if change_description:
task.set_comment(change_description)