search

qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad
MIT License
1.39k stars 529 forks source link

cannot deepcopy model #145

Open lzwhard opened 3 years ago

lzwhard commented 3 years ago

net = create_mobilenetv2_ssd_lite() import copy net_copy = copy.deepcopy(net)

error:

Traceback (most recent call last):
  File "run_ssd_live_demo.py", line 50, in <module>
    net2 = copy.deepcopy(net)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 186, in deepcopy
    y = _reconstruct(x, rv, 1, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 301, in _reconstruct
    state = deepcopy(state, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 155, in deepcopy
    y = copier(x, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 247, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 186, in deepcopy
    y = _reconstruct(x, rv, 1, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 301, in _reconstruct
    state = deepcopy(state, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 155, in deepcopy
    y = copier(x, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 247, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 155, in deepcopy
    y = copier(x, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 247, in _deepcopy_dict
    y[deepcopy(key, memo)] = deepcopy(value, memo)
  File "/workspace/liuzhenwei/venv/p3-d/lib/python3.5/copy.py", line 176, in deepcopy
    reductor = getattr(x, "__reduce_ex__", None)
TypeError: can't pickle PyCapsule objects