Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def on_epoch_end(self):
- 'Updates indexes after each epoch'
- self.indexes = np.arange(len(self.list_IDs))
- if self.shuffle == True:
- np.random.shuffle(self.indexes)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement