Skip to content

Commit 1c262cc

Browse files
authored
Update CIFAR_FS.py
1 parent 8823a06 commit 1c262cc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

data/CIFAR_FS.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -338,7 +338,7 @@ def sample_train_and_test_examples_for_novel_categories(
338338
Tnovel = []
339339
Exemplars = []
340340
assert((nTestNovel % nKnovel) == 0)
341-
nEvalExamplesPerClass = nTestNovel / nKnovel
341+
nEvalExamplesPerClass = int(nTestNovel / nKnovel)
342342

343343
for Knovel_idx in range(len(Knovel)):
344344
imd_ids = self.sampleImageIdsFrom(
@@ -427,4 +427,4 @@ def __call__(self, epoch=0):
427427
return self.get_iterator(epoch)
428428

429429
def __len__(self):
430-
return (self.epoch_size / self.batch_size)
430+
return int(self.epoch_size / self.batch_size)

0 commit comments

Comments
 (0)