pytorch - 随笔分类 - 一点飞鸿

ipdb调试pytorch的数据加载(dataset、 dataloader）出现 ValueError: I/O operation on closed file

摘要：修改：将dataloader中的num_works设置为0，之前是设置的4*显卡数量，是多个线程一起读，会出现错误阅读全文

posted @ 2023-07-26 17:56 一点飞鸿阅读(37) 评论(0) 推荐(0) 编辑

摘要：logits 与 pre-logits 阅读全文

posted @ 2023-07-19 15:47 一点飞鸿阅读(187) 评论(0) 推荐(0) 编辑

pytorch CrossEntropyLoss() 默认转换one-hot编码

摘要：import torchpredict = torch.randn((4,3))# crossentropyloss不需要predict的概率为1，predict为logits# predict = torch.nn.functional.softmax(predict,dim = 1)target 阅读全文

posted @ 2022-03-24 22:46 一点飞鸿阅读(456) 评论(0) 推荐(0) 编辑

随笔分类 - pytorch