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