Skip to content

Fix Denoising train on Torch 2.0#144

Open
anvuongb wants to merge 1 commit intoswz30:mainfrom
anvuongb:anvuongb-patch-1
Open

Fix Denoising train on Torch 2.0#144
anvuongb wants to merge 1 commit intoswz30:mainfrom
anvuongb:anvuongb-patch-1

Conversation

@anvuongb
Copy link
Copy Markdown

torch.sum() behavior changes in Torch 2.0, it will not work with a list, instead it only accept torch.Tensor as argument now. This is a simply fix to deal with that.

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