Change the BN to GN in resnet

[mobassir94 (Mobassir)]

thanks, i just solved the error,so after replacing bn with gn i had to do model = model.to(device)