image-classification pytorch

test