25 lines
672 B
TOML
25 lines
672 B
TOML
# 模型与数据, 其中 dataset_type 应当和 train_objective_params_list 对应起来
|
|
model_type = "cca2"
|
|
dataset_type = "objall"
|
|
dataset_dir = "F:/Datasets/MODatasetD"
|
|
# 训练参数
|
|
batch_size = 64
|
|
num_workers = 8
|
|
lr = 1e-4
|
|
epochs = 5
|
|
warmup_epochs = 1
|
|
# 其它
|
|
train_objective_params_list = [
|
|
"10x-0.25-1.0000", "10x-0.30-1.0000",
|
|
"20x-0.70-1.0000", "20x-0.80-1.0000",
|
|
"40x-0.65-1.0000", "100x-0.80-1.0000",
|
|
"100x-1.25-1.4730"
|
|
]
|
|
test_objective_params_list = [
|
|
"10x-0.25-1.0000", "10x-0.30-1.0000",
|
|
"20x-0.70-1.0000", "20x-0.80-1.0000",
|
|
"40x-0.65-1.0000", "100x-0.80-1.0000",
|
|
"100x-1.25-1.4730"
|
|
]
|
|
# 断点加载
|
|
checkpoint_load = true |