Closed feifeibear closed 2 years ago
The label tensor has a different shape as the loss, if you use batch splitting on the loss. Therefore I remove the cross cross_entropy from the lazy evaluation list.
The label tensor has a different shape as the loss, if you use batch splitting on the loss. Therefore I remove the cross cross_entropy from the lazy evaluation list.