your example was right, but when batch_size was 128, loss was nan, actually it occur error on 36,37th line in circleloss.py.
your example was right, but when batch_size was 128, loss was nan, actually it occur error on 36,37th line in circleloss.py.