Code

Improved model+EMA checkpointing#2292

Code

Merged

glenn-jocher merged 13 commits intomasterultralytics/yolov5:masterfrom resume_emaultralytics/yolov5:resume_emaCopy head branch name to clipboardFeb 25, 2021

+18-12

Commits

Commits on Feb 24, 2021

• Enhanced model+EMA checkpointing

  glenn-jocher

  committed
  643e466

• update

  glenn-jocher

  committed
  553b582

• bug fix

  glenn-jocher

  committed
  dc3ca84

• bug fix 2

  glenn-jocher

  committed
  554b2ed

• always save optimizer

  glenn-jocher

  committed
  8c677c1

• ema half

  glenn-jocher

  committed
  1dba401

• remove model.float()

  glenn-jocher

  committed
  02042ae

• model half

  glenn-jocher

  committed
  976002d

• carry ema/model in fp32

  glenn-jocher

  committed
  4d5ab3b

• rm model.float()

  glenn-jocher

  committed
  189bf63

• both to float always

  glenn-jocher

  committed
  bd4cb50

• cleanup

  glenn-jocher

  committed
  a09b8fa

• cleanup

  glenn-jocher

  committed
  dcc6a98