This commit is contained in:
@@ -163,6 +163,7 @@ class SimpleDataSet(Dataset):
|
||||
if self.mode == "train"
|
||||
else (idx + 1) % self.__len__()
|
||||
)
|
||||
continue
|
||||
|
||||
self.logger.error(
|
||||
"Failed after {} attempts: {}".format(
|
||||
|
||||
Reference in New Issue
Block a user