diff --git a/examples/contrib/cifar10/main.py b/examples/contrib/cifar10/main.py index 36851645b9c3..79055c9c0f17 100644 --- a/examples/contrib/cifar10/main.py +++ b/examples/contrib/cifar10/main.py @@ -348,7 +348,7 @@ def get_save_handler(config): return TrainsSaver(dirname=config["output_path"]) - return DiskSaver(config["output_path"]) + return DiskSaver(config["output_path"], require_empty=False) if __name__ == "__main__":