Browse Source

!1852 default dataset_sink_mode=True

Merge pull request !1852 from wukesong/modify_alexnet_sink_mode
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
05c1b36c51
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      example/alexnet_cifar10/eval.py
  2. +1
    -1
      example/alexnet_cifar10/train.py

+ 1
- 1
example/alexnet_cifar10/eval.py View File

@@ -36,7 +36,7 @@ if __name__ == "__main__":
parser.add_argument('--data_path', type=str, default="./", help='path where the dataset is saved')
parser.add_argument('--ckpt_path', type=str, default="./ckpt", help='if is test, must provide\
path where the trained ckpt file')
parser.add_argument('--dataset_sink_mode', type=bool, default=False, help='dataset_sink_mode is False or True')
parser.add_argument('--dataset_sink_mode', type=bool, default=True, help='dataset_sink_mode is False or True')
args = parser.parse_args()

context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)


+ 1
- 1
example/alexnet_cifar10/train.py View File

@@ -38,7 +38,7 @@ if __name__ == "__main__":
parser.add_argument('--data_path', type=str, default="./", help='path where the dataset is saved')
parser.add_argument('--ckpt_path', type=str, default="./ckpt", help='if is test, must provide\
path where the trained ckpt file')
parser.add_argument('--dataset_sink_mode', type=bool, default=False, help='dataset_sink_mode is False or True')
parser.add_argument('--dataset_sink_mode', type=bool, default=True, help='dataset_sink_mode is False or True')
args = parser.parse_args()

context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target)


Loading…
Cancel
Save