Isolate training of model