write a pytorch script for multiclass classification problem. Also include a validation step as well as data prep step Provide your response in exactly 6 paragraphs (separated nearby paragraphs with 3 aterisk symbols ***).
user