We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8823a06 commit 1c262ccCopy full SHA for 1c262cc
data/CIFAR_FS.py
@@ -338,7 +338,7 @@ def sample_train_and_test_examples_for_novel_categories(
338
Tnovel = []
339
Exemplars = []
340
assert((nTestNovel % nKnovel) == 0)
341
- nEvalExamplesPerClass = nTestNovel / nKnovel
+ nEvalExamplesPerClass = int(nTestNovel / nKnovel)
342
343
for Knovel_idx in range(len(Knovel)):
344
imd_ids = self.sampleImageIdsFrom(
@@ -427,4 +427,4 @@ def __call__(self, epoch=0):
427
return self.get_iterator(epoch)
428
429
def __len__(self):
430
- return (self.epoch_size / self.batch_size)
+ return int(self.epoch_size / self.batch_size)
0 commit comments