# Pt Train Classification Model Trains a classification model using a given dataset, optimizer, and number of epochs. ## Input | Name | Data type | |---|---| | model | Ptmodel | | train_loader | Ptdataloader | | optimizer | Ptoptimizer | | epochs | Int | | use_gpu | Boolean | | val_loader | Ptdataloader | ## Output | Data type | |---| | Ptmodel | | Tensor | | Tensor |