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/quantile_trainer.py
View File

@@ -675,7 +675,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
title="Final Training Plot", title="Final Training Plot",
series=f"Sample {actual_idx}", series=f"Sample {actual_idx}",
iteration=epoch, 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") fig2.savefig(f"sample_{actual_idx}_samples_plot.png")
@@ -683,7 +683,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
title="Final Training Samples Plot", title="Final Training Samples Plot",
series=f"Sample {actual_idx} samples", series=f"Sample {actual_idx} samples",
iteration=epoch, iteration=epoch,
image_path=f"sample_{actual_idx}_samples_plot.png", local_path=f"sample_{actual_idx}_samples_plot.png",
) )
plt.close() plt.close()

4
src/trainers/trainer.py
View File

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

4
src/training_scripts/non_autoregressive_quantiles.py
View File

@@ -2,7 +2,9 @@ from src.utils.clearml import ClearMLHelper
#### ClearML #### #### ClearML ####
clearml_helper = ClearMLHelper(project_name="Thesis/NAQR: Linear") clearml_helper = ClearMLHelper(project_name="Thesis/NAQR: Linear")
task = clearml_helper.get_task(task_name="NAQR: Linear + Load + PV + Wind + Net Position") task = clearml_helper.get_task(
task_name="NAQR: Linear + Load + PV + Wind + Net Position"
)
task.execute_remotely(queue_name="default", exit_process=True) task.execute_remotely(queue_name="default", exit_process=True)
from src.policies.PolicyEvaluator import PolicyEvaluator from src.policies.PolicyEvaluator import PolicyEvaluator