lichen/ieemoo-ai-contrast
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

更改

Browse Source
This commit is contained in:
lee
2025-06-18 11:30:55 +08:00
parent 180a41ae90
commit bf9604ec29
6 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
configs/distill.yml
View File

@ -51,7 +51,7 @@ data:
dataset: "imagenet" # 数据集名称（示例用，可替换为实际数据集）
train_batch_size: 128 # 训练批次大小
val_batch_size: 100 # 验证批次大小
num_workers: 4 # 数据加载线程数
num_workers: 16 # 数据加载线程数
data_train_dir: "../data_center/contrast_learning/data_base/train" # 训练数据集根目录
data_val_dir: "../data_center/contrast_learning/data_base/val" # 验证数据集根目录