# 模型与数据, 其中 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