DenseNet in fastai?

Full support for DenseNet seems to have been dropped in version 1.0 of fastai. However it is easy enough to use the Pytorch model directly.

from torchvision.models import densenet121

def dn121(pre): return children(densenet121(pre))[0]

learn = create_cnn(data, dn121)
2 Likes