# 模型与数据 model_type = "cca2" output_type = "distance" dataset_dir = "F:/Datasets/MODatasetD" # 训练参数 batch_size = 64 num_workers = 8 lr = 1e-4 patience = 5 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