Skip to content
Snippets Groups Projects
Commit 43c2c3bb authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 31883f30
Branches
No related tags found
No related merge requests found
......@@ -817,7 +817,7 @@ class CloudHeightNN:
print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
ckpt_manager.save()
if DISK_CACHE and epoch == 0:
if self.DISK_CACHE and epoch == 0:
f = open(cachepath, 'wb')
pickle.dump(self.in_mem_data_cache, f)
f.close()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment