Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: Improved training scripts and added updated pretrained weights #163

Merged
merged 28 commits into from
Nov 7, 2021

Conversation

frgfm
Copy link
Owner

@frgfm frgfm commented Nov 7, 2021

This PR introduces the following modifications:

  • deprecated LabelSmoothingCrossEntropy since PyTorch integrates it now
  • revamped Mixup
  • added wandb integration for experiment tracking
  • updated training script by switching augmentation to TrivialAugmentWide, and selecting more appropriate optimizers
  • updated pretrained weights of darknet19, cspdarknet53, cspdarknet53_mish, repvgg_a0, repvgg_a1, resnet50d
  • updated classification perf report

@frgfm frgfm added this to the 0.2.0 milestone Nov 7, 2021
@frgfm frgfm self-assigned this Nov 7, 2021
@codecov
Copy link

codecov bot commented Nov 7, 2021

Codecov Report

Merging #163 (e92d076) into master (cd3a624) will decrease coverage by 0.10%.
The diff coverage is 90.00%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master     #163      +/-   ##
==========================================
- Coverage   93.97%   93.86%   -0.11%     
==========================================
  Files          51       51              
  Lines        3005     2985      -20     
==========================================
- Hits         2824     2802      -22     
- Misses        181      183       +2     
Impacted Files Coverage Δ
holocron/models/darknetv2.py 100.00% <ø> (ø)
holocron/models/darknetv4.py 100.00% <ø> (ø)
holocron/models/repvgg.py 89.15% <ø> (ø)
holocron/models/resnet.py 94.82% <ø> (ø)
holocron/trainer/core.py 91.32% <89.28%> (-0.27%) ⬇️
holocron/nn/functional.py 92.25% <100.00%> (-0.19%) ⬇️
holocron/nn/modules/loss.py 90.90% <100.00%> (-3.00%) ⬇️

@frgfm frgfm merged commit b103d1b into master Nov 7, 2021
@frgfm frgfm deleted the training-recipes branch November 7, 2021 17:25
@frgfm frgfm added the type: enhancement New feature or request label Jan 9, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant