Fallback to 16-mixed AMP mode because of undiagnosed issue with `bf…

…16-mixed` Seemingly randomly, some jobs will fail with a 'TypeError: Got unsupported ScalarType BFloat16' error
creatis-myriad · Jun 27, 2024 · 8957e04 · 8957e04
1 parent 02542a9
commit 8957e04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/didactic/config/experiment/cardiac-multimodal-representation.yaml b/didactic/config/experiment/cardiac-multimodal-representation.yaml
@@ -15,7 +15,7 @@ test: True
 
 trainer:
   devices: 1
-  precision: bf16-mixed
+  precision: 16-mixed
 
 data:
   batch_size: 128