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,