报错原因:对于0维张量,无法使用索引访问。 将utils.py中的
self.losses[loss_name] = losses[loss_name].data[0]
修改为 self.losses[loss_name] = losses[loss_name].item() for i, loss_name in enumerate(losses.keys()): if loss_name not in self.losses: self.losses[loss_name] = losses[loss_name].item() #原始版本 #self.losses[loss_name] = losses[loss_name].data[0] else: self.losses[loss_name] += losses[loss_name].item() #原始版本 #self.losses[loss_name] += losses[loss_name].data[0]