swithmn1

[email protected]

Results 2 comments of


                                            swithmn1

graph.channel_dims这个函数有什么作用？

@wolfworld6 你好，我和你遇到同样的问题，请问解决了么

TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

output = self.model(data).cpu().numpy()；this is right!