feat(cluster): add ImageNet pre-trained model section and adjust batch sizes

This commit is contained in:
Shaun Zhou
2026-09-03 15:47:37 +08:00
parent 345dee50da
commit 4550fdfa8e
2 changed files with 5 additions and 3 deletions
+3 -3
View File
@@ -72,7 +72,7 @@ dataset:
- {name: PerImageStandardize} # 逐图 z-score,抹掉整体亮度/对比度
loader:
batch_size: 64
batch_size: 128
num_workers: 8
# method=simclr 时每个视图各自过一遍的随机增强(正样本对就是同一张图的两个视图)
@@ -95,7 +95,7 @@ simclr:
block: dw
feature_stages: null # tiny_conv 不是 nn.Sequential,取最终输出
epochs: 100
batch_size: 256
batch_size: 128
temperature: 0.2 # 越小越关注难负样本
amp: auto # CUDA 上开、MPS 上关
save_interval: 5 # 每 5 个 epoch 落一次盘,中途中断也能拿到权重
@@ -121,4 +121,4 @@ simclr:
# 输出:每簇软链多少张图 / 预览拼图放多少张
link_limit: 300
preview_count: 25
preview_count: 100