Open HZIH opened 11 months ago
The following is the args of the pretrained model provided:
Namespace(att_loss=False, aux_cls=False, batch_size_per_gpu=16, casme2_path='data/CASME2-RAW/', dataset='CASME2', decoder_depth=4, decoder_embed_dim=512, decoder_num_heads=4, depth=4, diag_att=False, dist_url='env://', distributed=True, drop_rate=0.0, ema_decay=0.997, embed_dim=512, enable_dino=True, epochs=200, gpu=0, has_decoder=True, img_size=224, local_crops_number=8, local_rank=0, lr=0.00025, min_lr=0.0, model_name='mae_vit_small_patch8', num_heads=4, num_workers=4, out_dim=128, output_dir='logs/mae_vit_small_patch8/CASME2-is224-p8-b16-ep200-0.5-mae-dino-2', patch_size=8, pretrained_checkpoint='', pretrained_encoder='dummy', rank=0, replace_ratio=0.5, resume='', samm_path='data/SAMM/', saveckp_freq=5, scheduler='cosine', seed=216, segmentation=False, teacher_temp=0.04, use_ema=False, use_fp16=False, warmup_teacher_temp=0.04, warmup_teacher_temp_epochs=0, weight_decay=0.0001, world_size=16)
I have 5 following problems
If you are afraid of bad use of your pretrained model, then just show us the other parts of the code or answer our questions, don't bury your head like an ostrich.