PyTorch Trainer Class
Project description
This is a PyTorch Model Trainer Class. It trains, validates input data. It also saves the model after training to avoid retraining. The class takes in 7 arguments i.e. device, epochs, model, criterion, optimizer, trainloader, valloader. It has 2 methods:: (1). model_train_validate(). This train, validate data.It also plots learning curves.Finally it saves the model to avoid retraining it.(2). load_model().This loads your saved model from the above function i.e. model_train_validate().
Change Log
0.0.3 (28/01/2023) - Third Release
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Close
Hashes for ClintonPytorchTrainer-0.0.4.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 84aafb0362d589729e6b8848d078a6c96012da77b9d867353596bfdff60a7d43 |
|
MD5 | 05dade2debcea8210052c6286e311207 |
|
BLAKE2b-256 | 389daddb9aec9d18f0e4806695a37b1ade393b2cfd8f7560d00088efc909bdfe |