unnecessary keras iterator for test data
keras iterator is created for train, val, and test data, but only used for train and val. Batch test data is not used at all but created. Remove creation of test data keras iterator from train stage.
-
adjust set_generators()
-
remove parameter self.test_set