Code

Custom optimizer#132

Code

Merged

klemen1999 merged 21 commits intomainluxonis/luxonis-train:mainfrom feat/yolov6n-optimizer-weight-initluxonis/luxonis-train:feat/yolov6n-optimizer-weight-initCopy head branch name to clipboardNov 19, 2024

+445-46

Commits

Commits on Nov 13, 2024

• Hardcode the warmup process and the weight initialization

  JSabadin

  committed
  792f6b9

Commits on Nov 14, 2024

• custom SGD idea

  JSabadin

  committed
  08f219e

• moving warmup_epochs to optim params

  JSabadin

  committed
  dfea01d

• remove warmup_epochs from config.py

  JSabadin

  committed
  58f7f3f

• remove comments

  JSabadin

  committed
  5c3a3a7

• Custom scheduler and optimizer classes

  JSabadin

  committed
  cefe3a6

• default params

  JSabadin

  committed
  00a2bd3

Commits on Nov 15, 2024

• code refactor; added training strategies

  JSabadin

  committed
  147c1ee

• use self.core

  JSabadin

  committed
  6e0c888

• type-check fix

  JSabadin

  committed
  c0bdbfc

• type-check fix

  JSabadin

  committed
  0c3bbbe

• add docs

  JSabadin

  committed
  dc1d468

Commits on Nov 16, 2024

• add docs and minor fix

  JSabadin

  committed
  9dfd8da

Commits on Nov 18, 2024

• fix failing tests

  JSabadin

  committed
  75096da

• modify the logic for assigning the optimizer and scheduler

  JSabadin

  committed
  23dda47

• fix failing test

  JSabadin

  committed
  1f16b46

Commits on Nov 19, 2024

• fix type-check error

  JSabadin

  committed
  d308e33

• type-check fix

  JSabadin

  committed
  f5965c7

• fix failing tests

  JSabadin

  committed
  093feec

• clamp kpts values

  JSabadin

  committed
  4c61f38

• initialize_weights configurable param

  JSabadin

  committed
  2951676