Closed reborm closed 1 year ago
请问下,在预测的时候加载模型时为什么一开始要使用torch.set_default_tensor_type(torch.cuda.HalfTensor),模型加载完成后又要使用torch.set_default_tensor_type(torch.cuda.FloatTensor)
请问下,在预测的时候加载模型时为什么一开始要使用torch.set_default_tensor_type(torch.cuda.HalfTensor),模型加载完成后又要使用torch.set_default_tensor_type(torch.cuda.FloatTensor)