ChenghaoYang comments

Repositories
Issues
Comments

Results 1 comments of


                                            ChenghaoYang

trafficstars

SOLUTION: IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

I changed the code as follows : > > ``` > if loss_name not in self.losses: > self.losses[loss_name] = losses[loss_name].data.cpu().numpy() > else: > self.losses[loss_name] += losses[loss_name].data.cpu().numpy() > However, there is...