-
Notifications
You must be signed in to change notification settings - Fork 1
Detailed Documentation
Marcin Paluch edited this page Apr 11, 2024
·
3 revisions
To train more complex models, you can write your own:
- Write your module
SI_Toolkit_ASF/ToolkitCustomization/Modules/[module_name].py
- Write a class that inherits from
tf.keras.Model - The class must have a
__init__(self)function and acall(self, x)function.xis a tensor containing the input to the network. Refer to https://keras.io/api/models/model/ for more information on how to write your own model.
- For training, follow the steps outlined above.
- Use your custom model by specifying
--net-name Custom-[module_name]-[class_name]