ResNet50
superkirill opened this issue · comments
superkirill commented
Could you also provide a permutation spec for ResNet50?
Lucca Zenóbio commented
done it. let me know if it works as expected, my GPU takes a while to train a resnet50.
use depth 52 as an arg
superkirill commented
Thanks! My concern is that it's not a very "canonical" implemetation with 4 block groups and varying numbers of channels, so it's hard to use it with official PyTorch models
Lucca Zenóbio commented
Yes, it should be easy to use the same logic with resnet from torchvision. Just have to remember to not use batchnorm
Santosh Sanjeev commented
Its not working with torchvisions ResNet50. Can you pls help me out
Thanks!
Yan Pan commented