Error in Feature visualization

[BalajiVenkatesan006]

After successfully training the models, I am trying to use this to visualize the feature space. but after a while the process is getting killed automatically. There is no clue in logs as well. Am I missing something ?

[08/13 16:52:22 d2.evaluation.evaluator]: Total inference pure compute time: 0:01:06 (0.020806 s / iter per device, on 1 devices) Killed