vishnu-m77 / DiffAPTOSC

Final Project for AMATH 495
1 stars 2 forks source link

Weighted Loss Wrapper #46

Closed Sehmimul closed 11 months ago

Sehmimul commented 11 months ago

https://github.com/vishnu-m77/DiffAPTOSC/blob/weighted_loss/src/diffusion.py#L211

Convert wrapper to a class. I think the wrapper is fine as it is, but if you think class would be cleaner, feel free to go for it.

vishnu-m77 commented 11 months ago

Resolved: https://github.com/vishnu-m77/DiffAPTOSC/commit/8179b3d914a0e24721e3a405d7a93477bb7c5b97 test branch.