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