I am always bemused by programmers that don’t use debuggers. It isn’t just about breakpoints and examining variables, but also being able to break into a process that has been running for an hour and issue something like:
(observation_ring*255).to(dtype=torch.uint8).view(128,16,4,3,128,128).permute(1,0,2,4,5,3).cpu().numpy().tofile('results/ring_128x128.rgb')