"test/vscode:/vscode.git/clone" did not exist on "2772751df7e9417d7b3cfe739d2e1e08602cf3bf"
Commit 39c38b2e authored by thomwolf's avatar thomwolf
Browse files

fix

parent dcddf498
......@@ -63,6 +63,9 @@ def load_pytorch_state_dict_in_tf2_model(tf_model, pt_state_dict, tf_inputs=None
raise e
# Adapt state dict - TODO remove this and update the AWS weights files instead
# Convert old format to new format if needed from a PyTorch state_dict
old_keys = []
new_keys = []
for key in pt_state_dict.keys():
new_key = None
if 'gamma' in key:
......
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