12 lines
522 B
Plaintext
12 lines
522 B
Plaintext
# StripNet small backbone (unfrozen) + Conv-MONA on last 2 stages.
|
|
ModelsCommonConfig.backbone = 'stripnet'
|
|
ModelsCommonConfig.baseline_mode = True
|
|
ModelsCommonConfig.init_gate = 0.7
|
|
ModelsCommonConfig.lrsclip_path = 'nn_models/LRSCLIP/DGTRS-CLIP-ViT-L-14.pt'
|
|
|
|
StripNetModelsConfig.stripnet_path = 'nn_models/STRIPNET/stripnet_s.pth'
|
|
StripNetModelsConfig.stripnet_freeze = False
|
|
StripNetModelsConfig.stripnet_mona_last_n_stages = 2
|
|
StripNetModelsConfig.stripnet_backbone_lr_factor = 0.1
|
|
StripNetModelsConfig.lora_rank = 4
|