Skip to content

GitLab

Unverified Commit fb4b9580 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

add missing `from_config` method

parent c1b65fc1
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
View file @ fb4b9580
...@@ -121,6 +121,9 @@ class BilinearGridSampler(tf.keras.layers.Layer): ...@@ -121,6 +121,9 @@ class BilinearGridSampler(tf.keras.layers.Layer):
def get_config(self): def get_config(self):
return self._config_dict return self._config_dict
@classmethod
def from_config(cls, config):
return cls(**config)
class PasteMasks(tf.keras.layers.Layer): class PasteMasks(tf.keras.layers.Layer):
...@@ -169,3 +172,7 @@ class PasteMasks(tf.keras.layers.Layer): ...@@ -169,3 +172,7 @@ class PasteMasks(tf.keras.layers.Layer):
def get_config(self): def get_config(self):
return self._config_dict return self._config_dict
@classmethod
def from_config(cls, config):
return cls(**config)
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