Add Automatic Mixed Precision option for training and evaluation. (#199)

This commit is contained in:
Alexander Soare
2024-05-20 18:57:54 +01:00
committed by GitHub
parent 2b270d085b
commit b6c216b590
4 changed files with 103 additions and 33 deletions

View File

@@ -10,6 +10,9 @@ hydra:
name: default
device: cuda # cpu
# `use_amp` determines whether to use Automatic Mixed Precision (AMP) for training and evaluation. With AMP,
# automatic gradient scaling is used.
use_amp: false
# `seed` is used for training (eg: model initialization, dataset shuffling)
# AND for the evaluation environments.
seed: ???