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

Fixed issue with clearml report image

Browse Source
This commit is contained in:
Victor Mylle
2024-04-19 15:09:55 +02:00
parent 2cdd2257a0
commit 0817f60e72
3 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/trainers/trainer.py
View File

@@ -367,7 +367,7 @@ class Trainer:
title="Final Training Plot",
series=f"Sample {actual_idx}",
iteration=epoch,
image_path=f"sample_{actual_idx}_plot.png",
local_path=f"sample_{actual_idx}_plot.png",
)
fig2.savefig(f"sample_{actual_idx}_samples_plot.png")
@@ -375,7 +375,7 @@ class Trainer:
title="Final Training Samples Plot",
series=f"Sample {actual_idx} samples",
iteration=epoch,
image_path=f"sample_{actual_idx}_samples_plot.png",
local_path=f"sample_{actual_idx}_samples_plot.png",
)
plt.close()