From b7b188bf54647313cfe07779fc5025b73273dc53 Mon Sep 17 00:00:00 2001
From: Krzysztof Chalupka <chalupka@fb.com>
Date: Tue, 2 Aug 2022 08:50:50 -0700
Subject: [PATCH] Fix train_stats.pdf: they now work by default

Summary: Before this diff, train_stats.py would not be created by default, EXCEPT when resuming training. This makes them appear from start.

Reviewed By: shapovalov

Differential Revision: D38320341

fbshipit-source-id: 8ea5b99ec81c377ae129f58e78dc2eaff94821ad
---
 projects/implicitron_trainer/impl/model_factory.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/implicitron_trainer/impl/model_factory.py b/projects/implicitron_trainer/impl/model_factory.py
index 9e9e7f01..728f7d57 100644
--- a/projects/implicitron_trainer/impl/model_factory.py
+++ b/projects/implicitron_trainer/impl/model_factory.py
@@ -159,6 +159,7 @@ class ImplicitronModelFactory(ModelFactoryBase):  # pyre-ignore [13]
         stats = Stats(
             # log_vars should be a list, but OmegaConf might load them as ListConfig
             list(log_vars),
+            plot_file=os.path.join(exp_dir, "train_stats.pdf"),
             visdom_env=visdom_env_charts,
             verbose=False,
             visdom_server=self.visdom_server,