Skip to content

Commit

Permalink
auto turn on amp
Browse files Browse the repository at this point in the history
  • Loading branch information
@stefanklut
stefanklut committed Mar 20, 2024
1 parent a6f0689 commit 32d58f9
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions configs/segmentation/baseline/baseline_general.yaml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,9 @@ MODEL:
SEM_SEG_HEAD:
NUM_CLASSES: 2

AMP_TRAIN:
ENABLED: True

WEIGHTS:

TRAIN:
Expand Down

0 comments on commit 32d58f9

Please sign in to comment.