Skip to content

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

Open
zinccat wants to merge 1 commit intoNielsRogge:masterfrom
zinccat:master
Open

fix type to comply with albumentation, fix shape in loss calculation#459
zinccat wants to merge 1 commit intoNielsRogge:masterfrom
zinccat:master

Conversation

@zinccat
Copy link

@zinccat zinccat commented Aug 8, 2024

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

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant