search

XiaohangZhan / face_recognition_framework

Multi-task face recognition framework based on PyTorch
MIT License
250 stars 55 forks source link

About Multi Task Learning #4

Open DuckJ opened 5 years ago

DuckJ commented 5 years ago

Great work!!! Can you give some examples for multi-task. I think the examples in 'experiments' are all single tasks. Is there anything wrong with my understanding?

XiaohangZhan commented 5 years ago

Yes, they are all single tasks. See TODO at the end of README.

DuckJ commented 5 years ago

Soga. If I change the config.yaml, add the batchsize,data_root ..., multi-task experiments should also be started . Is right?

XiaohangZhan commented 5 years ago

For example: train: rand_seed: 0 batch_size: [59, 5] # ratio ~ ms1m : webface loss_weight: [0.92, 0.08] # ratio ~ ms1m : webface base_lr: 0.3 lr_decay_scale: 0.1 lr_decay_steps: [8, 12] # need to increase a little bit because of more data max_epoch: 14 # need to increase a little bit because of more data momentum: 0.9 weight_decay: 0.0001 print_freq: 20 average_stats: 100 data_root:

I've not tried. Tell me if it does not work.

DuckJ commented 5 years ago

Ok. I will try it. Another question，I noticed that the input size of the model is 224. Did you try to compare the input performance of 112 dimensions?

XiaohangZhan commented 5 years ago

No. Pls have a try and tell me your observation. Many thanks.

XiaohangZhan commented 5 years ago

You may also try to change the feature dimension from 256 to 512.

DuckJ commented 5 years ago

For example: train: rand_seed: 0 batch_size: [59, 5] # ratio ~ ms1m : webface loss_weight: [0.92, 0.08] # ratio ~ ms1m : webface base_lr: 0.3 lr_decay_scale: 0.1 lr_decay_steps: [8, 12] # need to increase a little bit because of more data max_epoch: 14 # need to increase a little bit because of more data momentum: 0.9 weight_decay: 0.0001 print_freq: 20 average_stats: 100 data_root:

  • 'data/ms1m/images'
  • 'data/webface/images' data_list:
  • 'data/ms1m/list.txt'
  • 'data/webface/list.txt'

I've not tried. Tell me if it does not work.

I use the MS1M and a private dataset(2000 images), it works. But when I use the multi - gpus, error occurred. Even when I run single task, the multi-gpu process can not be started.

DuckJ commented 5 years ago

I have solved it. pytorch vision change to 0.3.1 can run