From cf4a082e11c919469f867f42d054f5e715bbf87e Mon Sep 17 00:00:00 2001
From: alexds9 <alexds9@gmail.com>
Date: Wed, 6 Sep 2023 23:04:12 +0300
Subject: [PATCH] 1. Fix to EMA samples arguments not respecting False value.

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 669c644..21d0466 100644
--- a/train.py
+++ b/train.py
@@ -917,10 +917,10 @@ def main(args):
 
             models_info = []
 
-            if (args.ema_decay_rate is None) or (args.ema_decay_sample_raw_training is not None):
+            if (args.ema_decay_rate is None) or args.ema_decay_sample_raw_training:
                 models_info.append({"is_ema": False, "swap_required": False})
 
-            if (args.ema_decay_rate is not None) and (args.ema_decay_sample_ema_model is not None):
+            if (args.ema_decay_rate is not None) and args.ema_decay_sample_ema_model:
                 models_info.append({"is_ema": True, "swap_required": args.ema_decay_device != device})
 
             for model_info in models_info: