From c07df3f499bcfb298884ea94662584cc7e666ffa Mon Sep 17 00:00:00 2001 From: Noam Shazeer Date: Tue, 9 Mar 2021 16:11:38 -0800 Subject: [PATCH] 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 --- mesh_tensorflow/transformer/transformer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mesh_tensorflow/transformer/transformer.py b/mesh_tensorflow/transformer/transformer.py index e3ab84f0..48a6a629 100644 --- 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