From a030ed0f3f268676ffa028a8a49e903b49dc14a0 Mon Sep 17 00:00:00 2001
From: sdtblck <sdtblck@gmail.com>
Date: Thu, 18 Jun 2020 03:56:38 +0100
Subject: [PATCH] add notification when spatial_augs are on
---
training/training_loop.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/training/training_loop.py b/training/training_loop.py
index f3a11bd..76a0279 100755
--- a/training/training_loop.py
+++ b/training/training_loop.py
@@ -299,7 +299,10 @@ def training_loop(
G.setup_weight_histograms(); D.setup_weight_histograms()
metrics = metric_base.MetricGroup(metric_arg_list)
+
print('Training for %d kimg...\n' % total_kimg)
+ if spatial_augmentations:
+ print('Augmenting fakes and reals\n')
dnnlib.RunContext.get().update('', cur_epoch=resume_kimg, max_epoch=total_kimg)
maintenance_time = dnnlib.RunContext.get().get_last_update_interval()
cur_nimg = int(resume_kimg * 1000)
--
GitLab