祥瑞哈哈哈

摘要： import torch import numpy as np import torch.nn as nn import torch.functional as F class LinearNet(nn.Module): def __init__(self): super(LinearNet, se 阅读全文

摘要： >>> torch.load('tensors.pt') # Load all tensors onto the CPU >>> torch.load('tensors.pt', map_location=torch.device('cpu')) # Load all tensors onto th 阅读全文