xiaohe9275的博客

PyTorch转Caffe模型这里使用的工具来自：PytorchToCaffe，主体代码如下：PytorchToCaffe-master|——Caffe| |——caffe.proto| |——caffe_pb2.py||

1、设置可训练参数PyTorch中可以使用torch.nn.Parameter()来设置可训练参数。Parameters类是Tensor类的子类，当它与Module类一起使用时，也就是将一个Parameters对象作为Module类的一个属性时，它们会自动添加到Module的参数列表中，例如在该Module类的parameters()迭代器中。import torchimport torch.nn as nnclass MyModule(nn.Module): def __init__(s

PyTorch GPU上训练的模型加载到CPUnet.load_state_dict(torch.load(model_file, map_location='cpu'))不设置map_location参数则会报错：RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are...