I’m doing something pretty trivial:
from fastai import *
from fastai.vision import *
from torchsummary import summary
from collections import OrderedDict
class MyMnistDS(Dataset):
...
train_ds = MyMnistDS(train_x, train_y)
valid_ds = MyMnistDS(valid_x, valid_y)
test_ds = MyMnistDS(test_x, test_y)
data = DataBunch.create(train_ds, valid_ds, test_ds)
len(data.train_ds), len(data.valid_ds) #len(data.test_ds)
#returns (51000, 9000)
type(data.train_ds), type(data.valid_ds), type(data.test_ds)
#returns (__main__.MyMnistDS, __main__.MyMnistDS, NoneType)
Notice the NoneType for data.test_ds. What happened to the type?
conda list | grep fastai
dataclasses 0.6 py_0 fastai
fastai 1.0.39 1 fastai
fastprogress 0.1.18 py_0 fastai
nvidia-ml-py3 7.352.0 py_0 fastai
regex 2018.01.10 py36h14c3975_1000 fastai
spacy 2.0.18 py36hf484d3e_1000 fastai
torchvision 0.2.1 py_0 fastai