add missing `from_config` method

fb4b9580 · srihari-humbarwadi · c1b65fc1 · fb4b9580
Unverified Commit fb4b9580 authored Oct 24, 2021 by srihari-humbarwadi
Show whitespace changes
Inline Side-by-side

Showing with 10 additions and 3 deletions

official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py ...projects/panoptic_maskrcnn/modeling/layers/paste_masks.py +10 -3

No files found.
--- a/official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py
+++ b/official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py
@@ -121,6 +121,9 @@ class BilinearGridSampler(tf.keras.layers.Layer):
  def get_config(self):
    return self._config_dict

+  @classmethod
+  def from_config(cls, config):
+    return cls(**config)


 class PasteMasks(tf.keras.layers.Layer):
@@ -169,3 +172,7 @@ class PasteMasks(tf.keras.layers.Layer):

  def get_config(self):
    return self._config_dict
+
+  @classmethod
+  def from_config(cls, config):
+    return cls(**config)