Closed fakhar-iqbal closed 1 year ago
This activs previously gave TensorImage but we can not run BSELossWithLogits on TensorImage implementation. In this repo, the output of your cell is in TensorBase, which is correct, but in actual it was throwing output as TensorImage
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
I hope this was a right commit
This activs previously gave TensorImage but we can not run BSELossWithLogits on TensorImage implementation. In this repo, the output of your cell is in TensorBase, which is correct, but in actual it was throwing output as TensorImage