提交 9235053e 编写于 作者: R root

refine train code

上级 3651cf68
......@@ -20,7 +20,6 @@ from utils.fp16_utils import create_master_params_grads, master_param_to_train_p
from utility import add_arguments, print_arguments
IMAGENET1000 = 1281167
#IMAGENET100 = 128660
parser = argparse.ArgumentParser(description=__doc__)
add_arg = functools.partial(add_arguments, argparser=parser)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册