Commit 82709fa4 authored by lr119628's avatar lr119628
Browse files

[update] added AdaNet to supported LayerList

parent 958f2180
......@@ -42,10 +42,11 @@ public class CNNArch2GluonLayerSupportChecker extends LayerSupportChecker {
supportedLayerList.add(AllPredefinedLayers.RESHAPE_NAME);
// supportedLayerList.add(AllPredefinedLayers.CROP_NAME);
supportedLayerList.add(AllPredefinedLayers.LARGE_MEMORY_NAME);
supportedLayerList.add(AllPredefinedLayers.EPISODIC_MEMORY_NAME);
supportedLayerList.add(AllPredefinedLayers.EPISODIC_MEMORY_NAME);
supportedLayerList.add(AllPredefinedLayers.DOT_PRODUCT_SELF_ATTENTION_NAME);
supportedLayerList.add(AllPredefinedLayers.LOAD_NETWORK_NAME);
supportedLayerList.add(AllPredefinedLayers.LAYERNORM_NAME);
supportedLayerList.add(AllPredefinedLayers.AdaNet_Name);
}
}
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