Remove deep copy

qubvel · qubvel · commit 692d3311af47 · 2020-11-27T00:53:03.000+03:00
diff --git a/segmentation_models_pytorch/encoders/resnet.py b/segmentation_models_pytorch/encoders/resnet.py
@@ -22,7 +22,7 @@
         number of feature tensors = 6 (one with same resolution as input and 5 downsampled),
         depth = 3 -> number of feature tensors = 4 (one with same resolution as input and 3 downsampled).
 """
-from copy import deepcopy
+from copy import copy
 
 import torch.nn as nn
 
@@ -107,7 +107,7 @@ def load_state_dict(self, state_dict, **kwargs):
     }
 }
 
-pretrained_settings = deepcopy(pretrained_settings)
+pretrained_settings = copy(pretrained_settings)
 for model_name, sources in new_settings.items():
     if model_name not in pretrained_settings:
         pretrained_settings[model_name] = {}

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`	`number of feature tensors = 6 (one with same resolution as input and 5 downsampled),`
`23`	`23`	`depth = 3 -> number of feature tensors = 4 (one with same resolution as input and 3 downsampled).`
`24`	`24`	`"""`
`25`		`-from copy import deepcopy`
	`25`	`+from copy import copy`
`26`	`26`
`27`	`27`	`import torch.nn as nn`
`28`	`28`
`@@ -107,7 +107,7 @@ def load_state_dict(self, state_dict, **kwargs):`
`107`	`107`	`}`
`108`	`108`	`}`
`109`	`109`
`110`		`-pretrained_settings = deepcopy(pretrained_settings)`
	`110`	`+pretrained_settings = copy(pretrained_settings)`
`111`	`111`	`for model_name, sources in new_settings.items():`
`112`	`112`	`if model_name not in pretrained_settings:`
`113`	`113`	`pretrained_settings[model_name] = {}`