CAT-KD
CAT-KD copied to clipboard
Configuration for ImageNet
Hi,
Thanks for the great repo.
Could you please specifically mention your configuration on ResNet34 -> ResNet18 on ImageNet? In the paper it says you use 2x2 pooling but the config has 4x4. Also in the config the batch size is set to 2?
Could you please share the config you used for imagenet including pooling size, beta parameter, and other details?
Thanks a lot! All the best :)
Please refer to the configuration in the paper. I set the batch size to 2 when I cleaned the code and forgot to recover it :( I will rectify it when I got time.