-
Ponku authored
* Moved more losses into classes * Added photometric loss * quick fix for ssim loss return value * added references * replaced with unsqueeze * renaming variables * add ref to consistency loss * made mask optional everywhere. generalised photometric displacement * smoothness typo * fixed flow channel selection bug * aligned with training script Co-authored-by:Joao Gomes <jdsgomes@fb.com>
2c1022e3