abhi-mosaic's picture
Add loss function and bool cast
f438eeb