Lots of debugging done. Euclidean distance + AP layer seem to be working in unison (both only test-time layers)