Make following changes. You are not using self.output so I have commented. This makes gradient None as you are not using it in forward pass ...
確定! 回上一頁