pytorch-classification
pytorch-classification copied to clipboard
Fix minor typo
Argument 'arch' passed to function reshape_model, but 'args.arch' is used to call it in the last 'else' clause.