diff --git a/train.py b/train.py index 23c7d0d..771fa95 100644 --- 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)