cluster prepping should be done outside model `forward`

[drinkingkazu]

Recent gnn model (in @francois-drielsma fork) has cluster prepping inside the forward function. This causes unnecessary data movement (cpu<=>gpu) and hence should be done beforehand (like in a parser).