trying to run compute only once. Training scales too hard...