Training script for the custom models
Nobody has claimed this yet.
Assessment
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Newbie friendliness
- 55/100
- Issue type
- Feature
- Clarity
- Mostly clear
- Activity status
- Active
- Tech stack
- python
- Domain
- machine-learning
Research direction
Start in scripts/train_custom.py, which does not exist yet, so check the blocking issues #20, #21 and #22 first to see where the Model A and Model B definitions and any data loading will come from. Build the CLI with --model, --epochs, --lr and --optimizer flags, then save the best weights to results/model_<X>.pt and the history to results/model_<X>_history.json. Done means python scripts/train_custom.py --model B runs to completion and writes both files.
Written by the indexing model from the issue text.
Description
Train Model A or B with the chosen optimizer.
Files: scripts/train_custom.py
Tasks
- CLI:
--model A|B --epochs 30 --lr 1e-3 --optimizer adam - Save best weights to
results/model_<X>.ptand history toresults/model_<X>_history.json
Done when
python scripts/train_custom.py --model B completes and writes both files.
Depends on
#20 #21 #22
Close with a commit or PR message containing Closes #<this issue>.
- Dominant language
- Jupyter Notebook
- Stars
- 0
- Forks
- 0
- Avg merge
- 1m
- Merged PRs (30d)
- 3
Getting set up
This project ships no dev container, Dockerfile or contributing guide, so setting up is up to you: start from its README, and see our first-contribution guide for the general steps.
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
More from VisionForgeCSE/CNN-For-Image-Classification
-
evaluation
Difficulty 2/5 1-3 hours Newbie friendliness 66/100
-
evaluation good first issue
Difficulty 2/5 1-3 hours Newbie friendliness 62/100
-
evaluation
Difficulty 2/5 1-3 hours Newbie friendliness 68/100
-
training
Difficulty 2/5 1-3 hours Newbie friendliness 68/100
-
Optimizer factoryOpentraining
Difficulty 2/5 1-3 hours Newbie friendliness 65/100
All issues in VisionForgeCSE/CNN-For-Image-Classification
Similar issues
-
question
Difficulty 2/5 1-3 hours Newbie friendliness 68/100
-
Kumo-Tabular weights unreachable behind restrictive egress proxies (Xet + plain HTTP both fail)Open
Difficulty 2/5 1-3 hours Newbie friendliness 67/100
NVIDIA/structured-data-models#1091 ·
Maintainers usually reply within 1 day
-
BUG: kmeans raises a weight-shape error on duplicate background rowsPossibly taken @guptaishaan claimed this today. Open
Difficulty 2/5 1-3 hours Newbie friendliness 78/100
Maintainers usually reply within 5 days
-
Difficulty 2/5 1-3 hours Newbie friendliness 77/100
google-deepmind/optax#1813 ·
Maintainers usually reply within 2 days
-
bot-triaged module: docs triaged
Difficulty 1/5 1-3 hours Newbie friendliness 88/100
Maintainers usually reply within 1 day