13 lines
283 B
TOML
13 lines
283 B
TOML
xlsx_files = [
|
|
"E:/Datasets/SparseFocusDataset/complete_image/complete_image/excel/02/224_dataset_little.xlsx"
|
|
]
|
|
|
|
batch_size = 64
|
|
learning_rate = 1e-4
|
|
epochs = 600
|
|
num_workers = 4
|
|
seed = 3407
|
|
|
|
# 为空表示从头训练
|
|
# 非空表示仅加载模型参数初始化
|
|
init_weight = "" |