Write pytorch code for a simple convolution network, the input size is 224,224,3 RGB image. Output has 10 classes. Use resnet51 as a backbone. Create a python class for the model. Make it compatible with GPU. 