mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-03 12:50:15 +08:00
added cpu + amp error
This commit is contained in:
1 parent
efbd1a1c18
commit
fcda19aa25
2 files changed
+3
-6
No files matched your search
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user