Fix imports

tensorflow_code/run_pretraining.py

@@ -19,8 +19,8 @@ from __future__ import division
 from __future__ import print_function
 
 import os
-import modeling
-import optimization
+from tensorflow_code import modeling
+from tensorflow_code import optimization
 import tensorflow as tf
 
 flags = tf.flags