Update train.py

elttaes · Jan 7, 2023 · 993fa02 · 993fa02
1 parent 3cd007a
commit 993fa02
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Fitness/flu/alphafold/train.py b/Fitness/flu/alphafold/train.py
@@ -25,7 +25,6 @@
 import tensorflow as tf
 from process_path import process_path
 file_save='./tmp/scratch_nopair'
-shutil.rmtree(file_save,True)
 writer = tf.summary.create_file_writer(file_save)
 model_params_dir='/home/public/bigdata/af2data/af2data'
 os.environ['TF_FORCE_UNIFIED_MEMORY'] = '1'
@@ -280,4 +279,4 @@ def main(_):
             tf.summary.scalar("Spearman", res, step=step)
 
 if __name__ == '__main__':
-    app.run(main)
+    app.run(main)