How to suppress this warning?

2

You are probably using DataParallel but returning a scalar in the network. You should return a batched output.

1 Like