gyr-kdgc
gyr-kdgc
## 问题描述 运行的代码如下(注释部分为torch.save(model.state_dict(), "model_torch16.pt")格式保存的模型,可以转换成功,但是现在想转换torch的script model,转换失败了): import torch from x2paddle.convert import pytorch2paddle from model import resnet #trained_model = resnet.resnet50(num_classes=2, include_top=True) #trained_model.load_state_dict(torch.load("model_torch16.pt")) trained_model=torch.jit.load("model_script_torch16.pt") tensor_shape=[1,3,224,224] input_tensor = torch.zeros_like(torch.empty(tuple(tensor_shape))) pytorch2paddle(module=trained_model, save_dir="./output22", jit_type="trace", input_examples=[input_tensor]) ##...
目前看文档好像只有coco、imagenet等官方数据集,数据集太大了,现在没法进行测试,自己的数据集也没法加载
bert示例运行报错 OP_REQUIRES failed at nccl_communicator.cc:116 : Internal: unhandled system error,请问如何解决?
2023-09-27 09:21:54.582250: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1351] Created TensorFlow device (/job:worker/replica:0/task:1/device:GPU:0 with 5211 MB memory) -> physical GPU (device: 0, name: Tesla V100-PCIE-16GB, pci bus id: 0000:0b:00.0, compute capability: 7.0) 2023-09-27 09:21:54.583242: I...