Closed saijo0404 closed 1 year ago
Hi, the student_netG
should be a supernet architecture.
Currently, we only support super_mobile_resnet_9blocks
. So I guess what you could do is follow the Fast GAN Compression tutorial to directly use the resnet_9blocks
teacher to train a `super_mobile_resnet_9blocks
once-for-all network, or train a mobile_resnet_9blocks
mobile teacher (or use our pre-trained one) and redistill a mobile_resnet_9blocks
student.
I suggest the first one.
ok, thank you so much. I try do this.
Hello, I used this bash to training "Once-for-all" Network, but I don't know what is there in the parameter --student_netG that I can use? My restore_teacher_G, restore_student_G and restore_D_path are use resnet_9blocks.