Sa-m's picture
Upload loss.py
a217b32