Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fine-tuning Resnet 50 model #188

Open
otsebriy opened this issue May 10, 2022 · 0 comments
Open

Fine-tuning Resnet 50 model #188

otsebriy opened this issue May 10, 2022 · 0 comments

Comments

@otsebriy
Copy link

I have tried to fine-tune the Resnet50_Final.pth on my own data with annotations, but the results look like the model was trained from the beginning without uploading pretrained weights. What could be a problem?
Current config below:

train:
  training_dataset: ml/output/face_detection/random350_frames_faces/annotations_training/annotations.txt
  network: resnet50 # Backbone network mobile0.25 or resnet50
  num_workers: 4 # Number of workers used in dataloading
  lr: 0.001 # initial learning rate
  momentum: 0.9 # the gradient of the past steps to determine the direction to go
  resume_net: ./weights/pretrained/Resnet50_Final.pth # path to pretrained weights
  resume_epoch: 0 # for pretrained weights, epoch on which training was ended
  weight_decay: 0.0005 # Weight decay for SGD
  gamma: 0.1 # Gamma update for SGD
  save_folder: ml/experiments/face_detection/weights/train_results

  cfg_re50:
    name: Resnet50
    min_sizes: [[16, 32], [64, 128], [256, 512]]
    steps: [8, 16, 32]
    variance: [0.1, 0.2]
    clip: false
    loc_weight: 2.0
    gpu_train: true
    batch_size: 6
    ngpu: 1
    epoch: 100
    decay1: 70
    decay2: 90
    image_size: 840
    pretrain: true
    return_layers:
      layer2: 1
      layer3: 2
      layer4: 3
    in_channel: 256
    out_channel: 256

Hmm, actually, I didn't try it with resume_epoch, but I will.
Maybe someone has other thoughts?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant