-
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.x
is 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]