pytorch collections.OrderedDict‘对象没有'to’属性

import torch
from collections import OrderedDict

def move_tensors_to_device(obj, device):
    if isinstance(obj, torch.Tensor):
        return obj.to(device)
    elif isinstance(obj, dict):
        new_dict = OrderedDict()
        for key, value in obj.items():
            new_dict[key] = move_tensors_to_device(value, device)
        return new_dict
    elif isinstance(obj, (list, tuple)):
        new_list = []
        for item in obj:
            new_list.append(move_tensors_to_device(item, device))
        return type(obj)(new_list)
    else:
        return obj

# 示例使用
model_dict = OrderedDict()
model_dict['conv1'] = torch.nn.Conv2d(3, 64, kernel_size=3)
model_dict['relu1'] = torch.nn.ReLU()
model_dict['conv2'] = torch.nn.Conv2d(64, 64, kernel_size=3)
model_dict['relu2'] = torch.nn.ReLU()

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model_dict = move_tensors_to_device(model_dict, device)