switched cpu amp order

This commit is contained in:
William Falcon
2019-07-25 11:08:31 -04:00
parent 0e42d28415
commit 4b04dc06d4
+4 -4
View File
@@ -438,14 +438,14 @@ class Trainer(TrainerIO):
# ON CPU
else:
# CHOOSE OPTIMIZER
# filter out the weights that were done on gpu so we can load on good old cpus
self.optimizers = model.configure_optimizers()
# run through amp wrapper
if self.use_amp:
raise MisconfigurationException('amp + cpu is not supported. Please use a GPU option')
# CHOOSE OPTIMIZER
# filter out the weights that were done on gpu so we can load on good old cpus
self.optimizers = model.configure_optimizers()
self.__run_pretrain_routine(model)
# return 1 when finished