2022年12月14日
【记录】torch.nn.CrossEntropyLoss报错及解决
摘要： 报错 在多分类语义分割问题中使用torch.nn.CrossEntropyLoss的时候，遇到的报错有： 1. Assertion `t >= 0 && t < n_classes` failed. 2. RuntimeError: Expected floating point type for 阅读全文
posted @ 2022-12-14 12:56 __一条秋刀鱼 阅读(951) 评论(0) 推荐(0)