fix default for plugins

2023-06-29 22:00:16 -04:00 · 2023-06-29 22:00:16 -04:00 · 1afaf59ec9
parent a72d455fc5
commit 1afaf59ec9
1 changed files with 4 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -749,7 +749,10 @@ def main(args):
        generate_samples(global_step=0, batch=batch)

    from plugins.plugins import load_plugin
-    plugins = [load_plugin(name) for name in args.plugins]
+    if args.plugins is not None:
+        plugins = [load_plugin(name) for name in args.plugins]
+    else:
+        plugins = []

    try:
        write_batch_schedule(args, log_folder, train_batch, epoch = 0)