Finetune on the pretrained model

Hi, thank you so much for sharing this great work. I'm thinking of doing some finetune one the 128^2 imagenet pretrained pkl model. However, I'm not sure how could I do that. For example, what syn_layers, head_layers,... should be and what arguments should I pass in to the train.py? Is it possible for you to provide an example on that?