雨哥这样写配置很方便
收藏
雨哥这样写配置很方便,学习了
__all__ = ['CONFIG', 'get']
CONFIG = {
'model_save_dir': "./output/zodiac",
'num_classes': 12,
'total_images': 7096,
'epochs': 20,
'batch_size': 256,
'image_shape': [3, 224, 224],
'LEARNING_RATE': {
'params': {
'lr': 0.00375
}
},
'OPTIMIZER': {
'params': {
'momentum': 0.9
},
'regularizer': {
'function': 'L2',
'factor': 0.000001
}
},
'LABEL_MAP': [
"ratt",
"ox",
"tiger",
"rabbit",
"dragon",
"snake",
"horse",
"goat",
"monkey",
"rooster",
"dog",
"pig",
]
}
def get(full_path):
for id, name in enumerate(full_path.split('.')):
if id == 0:
config = CONFIG
config = config[name]
return config
0
收藏
请登录后评论
这是通过什么API写的
学习了
确实方便很多
牛啊,收藏了
这样一直拿到最里面的参数,666
我这里看怎么字是虚的
你的电脑不行。
赶紧买个蓝天准系统,3090塞进去。
纯python
原来插件的原因,关了就好了
太狠了吧,直接把电脑毙掉了~~
想换电脑得找好借口