xinxiang7

pytorch交叉熵import torch# CrossEntropyLoss由LogSoftmax和Nllloss组成。# 输入和输出input_x = torch.tensor([[0.05, 0.95, 0], [0.1, 0.8, 0.1]])target_y = torch.tensor([1, 2])logsoftmax = torch.nn.LogSoftmax(dim = 1)logsoftmax_output = logsoftmax(input_x)nllloss

MMdetection3D安装步骤创建环境，并激活conda create -n open-mmlab python=3.7 -y# 激活conda activate open-mmlab安装pytorchconda install pytorch torchvision cudatoolkit=10.1 -c pytorchpip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://dow