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

Trying to reduce white space for saved matplotlib images

Browse Source
This commit is contained in:
Victor Mylle
2024-04-19 16:00:37 +02:00
parent 2680973baf
commit afa70fc3b3
17 changed files with 345 additions and 269 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/trainers/quantile_trainer.py
View File

@@ -670,7 +670,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
else:
print("Saving figs")
# fig to PIL image
fig.savefig(f"sample_{actual_idx}_plot.png")
fig.savefig(f"sample_{actual_idx}_plot.png", bbox_inches="tight")
task.get_logger().report_image(
title="Final Training Plot",
series=f"Sample {actual_idx}",
@@ -678,7 +678,9 @@ class NonAutoRegressiveQuantileRegression(Trainer):
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", bbox_inches="tight"
)
task.get_logger().report_image(
title="Final Training Samples Plot",
series=f"Sample {actual_idx} samples",