提交 70b39184 编写于 作者: Z zhengya01

add ce for human_pose_estimation

上级 99853c5b
......@@ -35,6 +35,7 @@ parser.add_argument('--model_name', type=str, default='cdssmNet',
parser.add_argument('--config', type=str, default='cdssm_base', help="The global config setting")
parser.add_argument('--enable_ce', action='store_true', help='If set, run the task with continuous evaluation logs.')
parser.add_argument('--epoch_num', type=int, help='Number of epoch')
parser.add_argument('--num_devices', type=int, default=1, help='Number of GPU devices')
DATA_DIR = os.path.join(os.path.expanduser('~'), '.cache/paddle/dataset')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册