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.2 (28/01/2023) - Second 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.2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8967c077d5cb7b09720a29270b7e2a6d3f90ceff31797540f5815d0f4f9914f3 |
|
MD5 | c1020c14d2244cc58187b32b15f44b72 |
|
BLAKE2b-256 | 75b9875b0f9e2de6c89be7d498e8396424b3eb3c1b18c15d8ed78ff0b740b11b |