代码拉取完成,页面将自动刷新
##################################################
# Training Config
##################################################
GPU = '0' # GPU
workers = 4 # number of Dataloader workers
epochs = 160 # number of epochs
batch_size = 12 # batch size
learning_rate = 1e-3 # initial learning rate
##################################################
# Model Config
##################################################
image_size = (448, 448) # size of training images
net = 'inception_mixed_6e' # feature extractor
num_attentions = 32 # number of attention maps
beta = 5e-2 # param for update feature centers
##################################################
# Dataset/Path Config
##################################################
tag = 'bird' # 'aircraft', 'bird', 'car', or 'dog'
# saving directory of .ckpt models
save_dir = './FGVC/CUB-200-2011/ckpt/'
model_name = 'model.ckpt'
log_name = 'train.log'
# checkpoint model for resume training
ckpt = False
# ckpt = save_dir + model_name
##################################################
# Eval Config
##################################################
visualize = True
eval_ckpt = save_dir + model_name
eval_savepath = './FGVC/CUB-200-2011/visualize/'
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。