added cpu + amp error

This commit is contained in:
William Falcon committed 2019-07-24 19:07:53 -04:00
1 parent efbd1a1c18
commit fcda19aa25
2 files changed
+3 -6

No files matched your search

+1 -5
View File
@@ -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)