simclr icon indicating copy to clipboard operation
simclr copied to clipboard

Command about fine-tune and semi-supervised learning on tf2

Open lucasliunju opened this issue 3 years ago • 2 comments

Could you please provide the command about finetune and semi-supervised learning on tf2. I just find the command on tf1 and I find that is not appliable for tf2

Thank you!

Yong

lucasliunju avatar Aug 14 '21 13:08 lucasliunju

I was successful using this command

python run.py --mode=train_then_eval --train_mode=finetune \
  --fine_tune_after_block=4 --zero_init_logits_layer=True \
  --global_bn=False --optimizer=momentum --learning_rate=0.1 --weight_decay=0.0 \
  --train_epochs=100 --train_batch_size=63 --warmup_epochs=0 \
  --dataset=imagenet --image_size=224 --eval_split=test --resnet_depth=152 \
  --checkpoint=pretrianed_model/r152_1x_sk0/saved_model --model_dir=pretrained_ft --use_tpu=False

wit543 avatar Feb 08 '22 00:02 wit543

@wit543 can you please share your directory structure, I am a bit unclear about the --checkpoint and --model_dir what is the directory contents

deepankarvarma avatar Jan 18 '24 09:01 deepankarvarma