diff --git a/npu_mnist_example/train_npu.py b/npu_mnist_example/train_npu.py index 5445dba..71bd625 100644 --- a/npu_mnist_example/train_npu.py +++ b/npu_mnist_example/train_npu.py @@ -53,10 +53,7 @@ if __name__ == "__main__": args, unknown = parser.parse_known_args() data_dir = '' pretrain_dir = '' - train_dir = '' - - #回传结果到openi - upload_openi() + train_dir = '' device_num = int(os.getenv('RANK_SIZE')) #使用单卡时