mirror of
https://github.com/wlkjyh/dianxuan.git
synced 2025-04-05 01:14:18 +08:00
38 lines
733 B
Python
38 lines
733 B
Python
"""
|
||
网络相关配置文件
|
||
"""
|
||
|
||
# 预训练权重
|
||
weight_model_path = './vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5'
|
||
|
||
# 输入图像大小，52*52*3
|
||
input_shape = (52,52,3)
|
||
|
||
# 如果不使用GPU训练，则传入列表，多个GPU则传入多个编号
|
||
gpu = []
|
||
# gpu = [0,1,2,3]
|
||
|
||
# 训练ecoph数
|
||
epochs = 3000
|
||
# 批次大小
|
||
batch_size = 128
|
||
|
||
# 学习率
|
||
lr = 1e-3
|
||
|
||
# 是否开启tensorboard
|
||
tensorboard = True
|
||
|
||
# tensorboard日志目录
|
||
tensorboard_log_dir = './logs'
|
||
|
||
# 验证集比例
|
||
valid_rate = 0.05
|
||
|
||
sample_path = './sample'
|
||
|
||
auto_best_checkpoint_path = './checkpoint/best.h5'
|
||
|
||
auto_epoch_checkpoint_path = './checkpoint/epoch_{epoch:03d}.h5'
|
||
|
||
model_save_path = './model.h5' |