From 288f61dfe2e9cd73023545e664b9973db877e61d Mon Sep 17 00:00:00 2001 From: GeiduanLiu <61015788+GeiduanLiu@users.noreply.github.com> Date: Tue, 31 Mar 2020 21:54:02 +0800 Subject: [PATCH] Update transform.py --- source/train/transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/train/transform.py b/source/train/transform.py index 66ff871aad..e9f4f020e0 100644 --- a/source/train/transform.py +++ b/source/train/transform.py @@ -50,7 +50,7 @@ def load_transform_node(graph): transform_node_pattern = "\ filter_type_\d+/matrix_\d+_\d+|\ filter_type_\d+/bias_\d+_\d+|\ -filter_type_\d+/idt_\d+_\d+\ +filter_type_\d+/idt_\d+_\d+|\ layer_\d+_type_\d+/matrix|\ layer_\d+_type_\d+/bias|\ layer_\d+_type_\d+/idt|\