Pytorch

THCUNN/BCECriterion.cu:42: Assertion `input >= 0. && input <= 1.` failed. THCUNN/BCECriterion.c...

迫于生计，复现YOLOv3，现主要记载实现流程。

Pytorch的数据加载主要依赖torch.utils.data.Dataset和torch.utils.data.DataLoader两个模块，可以完成如下格式的傻瓜式加载。 train_dataset = MyDataset(train...

训model嘛，损失函数总是要得。这里简单归纳一下常见的损失函数及其在Pytorch中的使用方法，以便查阅👨‍💻。 1. 交叉熵损失函数 nn.LogSoftmax() nn.NLLLoss() nn.CrossEntrop...