Fix missing classes argument to ResNet

[ovcharenkoo]

When running a benchmark on flowers dataset which only has 5 classes, I got a size mismatch error regardless of --num-classes I provided.

Apparently, classes argument was not forwarded to the ResNet initialization. Once fixed, I managed to successfully run training on the dataset with 5 classes.