added cpu + amp error

2026-09-25 14:00:39 +08:00 · 2019-07-24 19:07:53 -04:00
parent efbd1a1c18
commit fcda19aa25
2 changed files with 3 additions and 6 deletions
@@ -444,11 +444,7 @@ class Trainer(TrainerIO):

            # run through amp wrapper
            if self.use_amp:
-                # An example
-                model, optimizers = amp.initialize(
-                    model, self.optimizers, opt_level=self.amp_level,
-                )
-                self.optimizers = optimizers
+                raise MisconfigurationException('amp + cpu is not supported. Please use a GPU option')

            self.__run_pretrain_routine(model)