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