本文整理汇总了Python中common.fit.add_fit_args方法的典型用法代码示例。如果您正苦于以下问题:Python fit.add_fit_args方法的具体用法?Python fit.add_fit_args怎么用?Python fit.add_fit_args使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类common.fit
的用法示例。
在下文中一共展示了fit.add_fit_args方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: prepare_args
# 需要导入模块: from common import fit [as 别名]
# 或者: from common.fit import add_fit_args [as 别名]
def prepare_args():
# parse args
parser = argparse.ArgumentParser(description="train cifar10",
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
fit.add_fit_args(parser)
data.add_data_args(parser)
data.add_data_aug_args(parser)
# uncomment to set standard cifar augmentations
# set_cifar_aug(parser)
parser.set_defaults(
# network
network = 'resnet',
num_layers = 110,
# data
data_train = train_fname,
data_val = val_fname,
num_classes = 10,
num_examples = 50000,
image_shape = '3,28,28',
pad_size = 4,
# train
batch_size = 128,
num_epochs = 300,
lr = .05,
lr_step_epochs = '200,250',
)
return parser.parse_args(), fit, data