Skip to content

GitLab

Commit d3a3a035 authored by thomwolf's avatar thomwolf
Browse files

clean up cache after conversion

parent a84adddd
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
pytorch_transformers/convert_pytorch_checkpoint_to_tf2.py
View file @ d3a3a035
...@@ -126,6 +126,8 @@ def convert_all_pt_checkpoints_to_tf(args_model_type, tf_dump_path, compare_with ...@@ -126,6 +126,8 @@ def convert_all_pt_checkpoints_to_tf(args_model_type, tf_dump_path, compare_with
config_file, config_file,
os.path.join(tf_dump_path, shortcut_name + '-tf_model.h5'), os.path.join(tf_dump_path, shortcut_name + '-tf_model.h5'),
compare_with_pt_model=compare_with_pt_model) compare_with_pt_model=compare_with_pt_model)
os.remove(config_file)
os.remove(model_file)
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment