pytorch拷贝模型

  • 对于模型中部分模块的复制需求,可以放心复制Module下的成员,然后再组装
import copy
# model = model.to('cpu')
# 如果模型当前在GPU上而你对此不放心的话
model_copy = copy.deepcopy(model)

你可能感兴趣的:(pytorch,模型,拷贝)