Yes, training may happen pretty fast in 2D, but then applied to the 3D to final visualizations. But this is just an assumption. I haven’t tried this approach yet.
https://rednuht.org/genetic_cars_2/ does something like this. You can turn off visualizations and let it run for a while then turn visualizations back on or replay the best result.