Add EMA and model checkpointing (#49)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ayush Chaurasia 2022-11-19 23:37:26 +05:30 committed by GitHub
parent 27d6545117
commit 4291b9c31c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 55 additions and 21 deletions

View file

@ -1,5 +1,4 @@
import os
from pathlib import Path
import numpy as np
import torch