remove unacessary files
fix loss name and axis