replace os.rename with shutil.move (fix #216)

tensorpack/train/monitor.py

@@ -4,6 +4,7 @@
 # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
 
 import os
+import shutil
 import operator
 from collections import defaultdict
 import six
@@ -203,7 +204,7 @@ class JSONWriter(TrainingMonitor):
         try:
             with open(tmp_filename, 'w') as f:
                 json.dump(self._stats, f)
-            os.rename(tmp_filename, self._fname)
+            shutil.move(tmp_filename, self._fname)
         except IOError:  # disk error sometimes..
             logger.exception("Exception in StatHolder.finalize()!")
 

tensorpack/user_ops/__init__.py

@@ -18,4 +18,6 @@ if ret != 0:
     print("Failed to compile user ops!")
 else:
     recv_mod = tf.load_op_library(os.path.join(file_dir, 'zmq_recv_op.so'))
+    # TODO trigger recompile when load fails
     zmq_recv = recv_mod.zmq_recv
+