Allow init_from_checkpoint to accept a list of pairs, so as to enable…

… initialization of multiple variables in the graph from the same variable in the checkpoint. PiperOrigin-RevId: 361920642
tensorflow · Mar 19, 2021 · ee9fa2e · ee9fa2e
1 parent 5d16bca
commit ee9fa2e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mesh_tensorflow/transformer/transformer.py b/mesh_tensorflow/transformer/transformer.py
@@ -45,6 +45,7 @@
 
 import json
 import math
+import re
 
 import gin
 import mesh_tensorflow as mtf