Commit 36923897 authored by danielkisov's avatar danielkisov
Browse files

Update README.md

parent f39a5095
...@@ -26,14 +26,14 @@ In order to visualize attention from an attention network, the data from this la ...@@ -26,14 +26,14 @@ In order to visualize attention from an attention network, the data from this la
* Place your custom layer python file inside the custom_layers folder (for the appropriate backend) and include the three functions in its class as shown in the example below * Place your custom layer python file inside the custom_layers folder (for the appropriate backend) and include the three functions in its class as shown in the example below
``` ```
# A method to return a dictionary with # A method to return a dictionary with
# the parameters of the layer as keys and their data type as value # the names of the parameters of the layer (string) as keys and their data type (data type object) as value
def get_parameters(self): def get_parameters(self):
return {'alpha': int, 'ones': int} return {'alpha': int, 'ones': int}
# A method to print a dictionary with # A method to print a dictionary with
# the parameters of the layer as keys and their data type as value # the names of the parameters of the layer (string) as keys and their data type (string) as value
def print_parameters(self): def print_parameters(self):
print({'alpha': 'int', 'ones': 'int'}) print({'alpha': 'int', 'ones': 'int'})
......
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