'Conv2d' object has no attribute 'weight'

I’m working on Zach’s walkthrough, and on this notebook


Running on a p2.xlarge deep learning ubuntu ami, with the following config https://gist.github.com/foobar8675/f6f352f9ddada60a45da6257b4ba949a with these relevant libraries

    - fastai2==0.0.11
    - fastcore==0.1.14
    - fastprogress==0.2.2
    - torch==1.4.0
    - torchvision==0.5.0

I’m getting the following error when calling

def create_gen_learner():
  return unet_learner(dls_gen, bbone, loss_func=loss_gen,
learn_gen = create_gen_learner()

and narrowed it down to module.py line 576

    def __getattr__(self, name):
        if '_parameters' in self.__dict__:
            _parameters = self.__dict__['_parameters']
            if name in _parameters:
                return _parameters[name]
        if '_buffers' in self.__dict__:
            _buffers = self.__dict__['_buffers']
            if name in _buffers:
                return _buffers[name]
        if '_modules' in self.__dict__:
            modules = self.__dict__['_modules']
            if name in modules:
                return modules[name]
# this line
        raise AttributeError("'{}' object has no attribute '{}'".format(

#error msg
'Conv2d' object has no attribute 'weight'

It works fine in colab but don’t know why it does not work on my aws instance. Any suggestions would be welcome.