Bug fix

Fix: immediate output of accuracy loss