fix tpu tests

This commit is contained in:
William Falcon 2020-06-30 17:20:35 -04:00
parent fc26078e39
commit 04e68f022f
1 changed files with 1 additions and 1 deletions

View File

@ -1193,7 +1193,7 @@ class Trainer(
# attempt to load weights from a spawn
path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt')
test_model = self.model
if os.path.exists(path):
if os.path.exists(path) and self.on_colab_kaggle:
test_model = self.load_spawn_weights(self.model)
self.fit(test_model)