network module
CATXHaikuNetwork
Bases: hk.Module
An interface for implementing a custom Haiku neural network.
__init__(depth)
abstractmethod
Specify the network architecture.
The dimension of the network output layer must be; 2**(depth+1).
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
depth |
int
|
the depth at which the network will be used in the tree. |
required |