yes, it raises an error now. you can check with a simple MNIST convnet:

class Net(nn.Container):
    def __init__(self):
        super(Net, self).__init__()
        self.conv1 = nn.Conv2d(1, 10, 5)
        self.pool1 = nn.MaxPool2d(2,2)
        self.conv2 = nn.Conv2d(10, 20, 5)
        self.pool2 = nn.MaxPool2d(2, 2)
        self.fc1   = nn.Linear(320, 50)
        self.fc2   = nn.Linear(50, 10)
        self.relu  = nn.ReLU()
        self.softmax = nn.LogSoftmax()

    def forward(self, x):
        x = self.relu(self.pool1(self.conv1(x)))
        x = self.relu(self.pool2(self.conv2(x)))
        x = x.view(-1, 320)
        x = self.relu(self.fc1(x))
        x = self.relu(self.fc2(x))
        return self.softmax(x)

model = Net()

import copy

m2 = copy.deepcopy(model)

---------------------------------------------------------------------------
NotImplementedError                       Traceback (most recent call last)
<ipython-input-7-4cf3c50bce8e> in <module>()
----> 1 m2 = copy.deepcopy(model)

/Users/soumith/anaconda/lib/python2.7/copy.pyc in deepcopy(x, memo, _nil)
    188                             raise Error(
    189                                 "un(deep)copyable object of type %s" % cls)
--> 190                 y = _reconstruct(x, rv, 1, memo)
    191
    192     memo[d] = y

/Users/soumith/anaconda/lib/python2.7/copy.pyc in _reconstruct(x, info, deep, memo)
    332     if state is not None:
    333         if deep:
--> 334             state = deepcopy(state, memo)
    335         if hasattr(y, '__setstate__'):
    336             y.__setstate__(state)

/Users/soumith/anaconda/lib/python2.7/copy.pyc in deepcopy(x, memo, _nil)
    161     copier = _deepcopy_dispatch.get(cls)
    162     if copier:
--> 163         y = copier(x, memo)
    164     else:
    165         try:

/Users/soumith/anaconda/lib/python2.7/copy.pyc in _deepcopy_dict(x, memo)
    255     memo[id(x)] = y
    256     for key, value in x.iteritems():
--> 257         y[deepcopy(key, memo)] = deepcopy(value, memo)
    258     return y
    259 d[dict] = _deepcopy_dict

/Users/soumith/anaconda/lib/python2.7/copy.pyc in deepcopy(x, memo, _nil)
    170             y = _deepcopy_atomic(x, memo)
    171         else:
--> 172             copier = getattr(x, "__deepcopy__", None)
    173             if copier:
    174                 y = copier(memo)

/Users/soumith/code/pytorch/torch/nn/backends/backend.pyc in __getattr__(self, name)
      7         fn = self.function_classes.get(name)
      8         if fn is None:
----> 9             raise NotImplementedError
     10         return fn
     11

NotImplementedError:

add copy.deepcopy support to nn.Container models #415

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions