https://github.com/aidinhamedi/pytorch-img-classification-trainer
A Pytorch image classification training system, easy to use (relatively 😅). Supports Tensorboard logging and techniques like "Gradient centralization", "Adaptive gradient clipping" and etc...
https://github.com/aidinhamedi/pytorch-img-classification-trainer
Science Score: 26.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
○CITATION.cff file
-
✓codemeta.json file
Found codemeta.json file -
✓.zenodo.json file
Found .zenodo.json file -
○DOI references
-
○Academic publication links
-
○Committers with academic emails
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (15.3%) to scientific vocabulary
Repository
A Pytorch image classification training system, easy to use (relatively 😅). Supports Tensorboard logging and techniques like "Gradient centralization", "Adaptive gradient clipping" and etc...
Basic Info
Statistics
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
- Releases: 0
Metadata Files
README.md
Pytorch Image Classification Trainer
A Pytorch image classification training system, easy to use (relatively 😅) Supports Tensorboard logging and techniques like Gradient centralization, Adaptive gradient clipping and etc...
🚀 Getting started
Step 1: Clone the repository
bash
git clone https://github.com/AidinHamedi/Pytorch-Img-Classification-Trainer.git
Step 2: Install the requirements
This repo uses uv to manage it dependencies.
bash
uv sync
Step 3: Check out the example
There is an example already made that uses this system to run experiments (hyper parameters tuning), The experiments params are set in the ./expers.toml file and the experiment runner is ./run_expers.py.
The experiment runner will run the train function in ./train_exper.py with the experiment params as the arg in that function you can set what the params do and etc...
In end you will see that the fit function of the Training_Engine is being called and here the magic happens 😊.
📚 Documentation
Training Engine
You can access the main fit function from ./Training_Engine/trainer.py file.
The fit function takes the following required arguments:
model: The model to be trained.train_dataloader: The training data loader. (DynamicArg)test_dataloader: The test data loader. (DynamicArg)optimizer: The optimizer to be used for training.loss_fn: The loss function to be used for training.
And done all of the other args are optional and used setting mixed precision and etc...
I think you have noticed that the train_dataloader and test_dataloader are DynamicArg so what is a DynamicArg?
DynamicArg
A DynamicArg is a special type of argument that allows you to pass a function as an generator and outputs a value based on the the environment.
Like for example you make the train_dataloader DynamicArg return a pytorch dataloader that adjusts the augmentation amount based on the epoch count.
Its not that complicated just by looking at the code you will understand it.
You can import dynamic args from ./Training_Engine/Utils/Base/dynamic_args.py.
Utils
There are some utils with the training engine that you can use for like loading the images and etc...
./Training_Engine/Utils/Data/data_loader.py: A utility for loading images from a directory or making a pytorch dataset for loading large datasets on the fly../Training_Engine/Utils/Data/normalization.py: A utility for normalizing images and getting class weighting.
📷 Example Output
📝 License
Copyright (c) 2025 Aidin Hamedi This software is released under the MIT License. https://opensource.org/licenses/MIT
Owner
- Name: Aidin
- Login: AidinHamedi
- Kind: user
- Repositories: 1
- Profile: https://github.com/AidinHamedi
Segmentation fault
GitHub Events
Total
- Watch event: 2
- Public event: 1
- Push event: 10
Last Year
- Watch event: 2
- Public event: 1
- Push event: 10
Issues and Pull Requests
Last synced: about 1 year ago
All Time
- Total issues: 0
- Total pull requests: 0
- Average time to close issues: N/A
- Average time to close pull requests: N/A
- Total issue authors: 0
- Total pull request authors: 0
- Average comments per issue: 0
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0
Past Year
- Issues: 0
- Pull requests: 0
- Average time to close issues: N/A
- Average time to close pull requests: N/A
- Issue authors: 0
- Pull request authors: 0
- Average comments per issue: 0
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0