fix resolution in training script