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.3.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4e3be166843f24f432330dcd6d9328ab99421a7e72578758e28ae7a1ba0f4a58 |
|
MD5 | 7b47282bed02a34eed2b168ca5c0906f |
|
BLAKE2b-256 | aa99eef06b463de47ad1e1973b16e7d653883a325d0073bf4c7c3bebedd90e92 |