From 41cad3fa2823fd17e4e7a2ac13be73519d59c862 Mon Sep 17 00:00:00 2001
From: Gal Shitrit <46653045+galsh17@users.noreply.github.com>
Date: Mon, 29 Jun 2020 09:26:50 +0300
Subject: [PATCH] Update noveou_train_netvlad_v3.py

fix keras generator bug
---
 noveou_train_netvlad_v3.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/noveou_train_netvlad_v3.py b/noveou_train_netvlad_v3.py
index 2aa8d16..917742a 100644
--- a/noveou_train_netvlad_v3.py
+++ b/noveou_train_netvlad_v3.py
@@ -98,7 +98,7 @@ def __getitem__(self, idx):
        #TODO: Can return another number (sample_weight) for the sample. Which can be judge say by GMS matcher. If we see higher matches amongst +ve set ==> we have good positive samples,
 
 
-    def on_epoch_start(self):
+    def on_epoch_end(self):
         N = self.refresh_data_after_n_epochs
         #print '-----on_epoch_start, self.epoch=', self.epoch, '\tself.refresh_data_after_n_epochs=', self.refresh_data_after_n_epochs