|
|
|
@ -67,7 +67,7 @@ if __name__ == '__main__':
|
|
|
|
|
parser.add_argument("--distribute", type=bool, default=False, help="Run distribute, default is false.")
|
|
|
|
|
parser.add_argument("--device_id", type=int, default=0, help="Device id, default is 0.")
|
|
|
|
|
parser.add_argument("--device_num", type=int, default=1, help="Use device nums, default is 1.")
|
|
|
|
|
parser.add_argument("--mode", type=str, default="sink", help="Run sink mode or non-sink mode, default is sink")
|
|
|
|
|
parser.add_argument("--mode", type=str, default="sink", help="Run sink mode or not, default is sink")
|
|
|
|
|
parser.add_argument("--epoch_size", type=int, default=10, help="Epoch size, default is 10")
|
|
|
|
|
parser.add_argument("--batch_size", type=int, default=32, help="Batch size, default is 32.")
|
|
|
|
|
parser.add_argument("--checkpoint_path", type=str, default="", help="Checkpoint file path")
|
|
|
|
|