Finetuning the public models

[fabianwindbacher]

Hi, very cool repo!

I am trying to fine-tune one of your public models (e.g. metrabs_mob3l_y4t) on a new dataset. Since these are the packaged multi-person models, I extracted the crop model itself.

However, using the checkpoint system implemented in main.py, it seems like the weights are not copied. log_detailed.txt:

WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).variables.0
WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).variables.1
...
WARNING:tensorflow:Value in checkpoint could not be found in the restored object: (root).variables.265

Can you point me in the right direction?