Transformers-Tutorials icon indicating copy to clipboard operation
Transformers-Tutorials copied to clipboard
verified publisher icon NielsRogge

fix type to comply with albumentation, fix shape in loss calculation

Open zinccat opened this issue 1 year ago • 1 comments

changed the normalization constant to list[float], removed squeeze in loss calculation step so a full batch can be trained without error

zinccat avatar Aug 08 '24 04:08 zinccat

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

review-notebook-app[bot] avatar Aug 08 '24 04:08 review-notebook-app[bot]