feat(configs): 新增 resnet101 的自监督聚类配置
resnet101 此前不在 TorchvisionBackbone 的支持列表里,先补进 _SUPPORTED (out_channels 2048),随后被既有的参数化测试自动覆盖:能前向、声明维度与实际一致。 configs/cluster/film_simclr_resnet101.yaml 仿照 resnet18/50 两份,只改骨架与 batch_size,其余超参一致以便横向对比: - 42.5M 参数(resnet18 11.2M / resnet50 23.5M),layer2+3 仍是 1536 维, 与 resnet50 相同,差别在深度(layer3 有 23 个 block 而非 6 个) - batch_size 取 128:MPS 上 resnet50 在 1024 已 OOM,resnet101 约重 1.8 倍故再降一档; 注释里说明该数字来自 MPS,16 GB 独显上可往上调 - pretrained 首次使用会下载约 171 MB 权重 Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5
parent
c9e7e4104c
commit
6e543660ce
@@ -12,6 +12,7 @@ _SUPPORTED = {
|
||||
"resnet18": 512,
|
||||
"resnet34": 512,
|
||||
"resnet50": 2048,
|
||||
"resnet101": 2048,
|
||||
"mobilenet_v3_small": 576,
|
||||
"mobilenet_v3_large": 960,
|
||||
"shufflenet_v2_x1_0": 1024,
|
||||
|
||||
Reference in New Issue
Block a user