&`i(ddlmZmZmZddlmZddlmZmZm Z e d deeee fde fdZ eddeee efde fd Z ed Z edd ZdS))CallableOptionalUnion) DeveloperAPI)try_import_jax try_import_tftry_import_torchNtfname frameworkct|r|St|tr|n|}|dkrl|dvrdSt \}}t ||d}||S|dvr|jS|dkr|jS|dkr|jS|dkr|j S|dkr|j Sn|d kr^|dvrdSt\}}|dvr |j j S|dkr |j jS|dkr |j jS|dkr |j jSnS|d vsJd ||dvrdSt%\}}} t |j |d}||St'd ||) aReturns a framework specific activation function, given a name string. Args: name: One of "relu" (default), "tanh", "elu", "swish" (or "silu", which is the same), or "linear" (same as None). framework: One of "jax", "tf|tf2" or "torch". Returns: A framework-specific activtion function. e.g. tf.nn.tanh or torch.nn.ReLU. None if name in ["linear", None]. Raises: ValueError: If name is an unknown activation function. torch)linearNN)swishsilurelutanhelusoftmaxjaxr tf2Unsupported framework `{}`!)Unknown activation ({}) for framework={}!)callable isinstancestrlowerr getattrSiLUReLUTanhELUSoftmaxrnnrr hard_tanhrformatr ValueError) r r name_lower_r%fnrtf1r tfvs j/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/models/utils.pyget_activation_fnr/s(~~ !+D#!6!6@DJG ) ) )4 ""2Rt $ $ >I * * *7N 6 ! !7N 6 ! !7N 5 6M 9 $ $: % e   ) ) )4!!Q * * *6<    6;  6 ! !6# # 5 6: !M)))+H+O+O , , ))) ) ) )4$ RRUJ - - >I 3::4KK  rct|s||S|dkrt|tr|n|}t \}}|dst d|dt|j|d}||St d|d|dkrDt\}}}t|j j |d}||St d |d dS) aReturns the framework-specific initializer class or function. This function relies fully on the specified initializer classes and functions in the frameworks `torch` and `tf2` (see for `torch` https://pytorch.org/docs/stable/nn.init.html and for `tf2` see https://www.tensorflow.org/api_docs/python/tf/keras/initializers). Note, for framework `torch` the in-place initializers are needed, i.e. names should end with an underscore `_`, e.g. `glorot_uniform_`. Args: name: Name of the initializer class or function in one of the two supported frameworks, i.e. `torch` or `tf2`. framework: The framework string, either `torch or `tf2`. Returns: A framework-specific function or class defining an initializer to be used for network initialization, Raises: `ValueError` if the `name` is neither class or function in the specified `framework`. Raises also a `ValueError`, if `name` does not define an in-place initializer for framework `torch`. Nrr*zNot an in-place initializer: Torch weight initializers need to be provided as their in-place version, i.e. + '_'. See https://pytorch.org/docs/stable/nn.init.html. User provided .zUnknown initializer name: z$ is not a method in `torch.nn.init`!rzUnknown initializer: z1 is not a initializer in `tf.keras.initializers`!) rrrrr endswithr(rinitrkeras initializers)r r r)r*r%r+r s r.get_initializer_fnr7WsS8~~ G%/c%:%:DTZZ\\\  ""2""3'' )"& ))) RWj$ / / >I#Z###  e   !??2qRX*D$ 7 7 >I++++   r0cjdddgdgdddgdgdddgdgg}dddgdgdddgdgdddgdgg}dddgdgdddgdgd ddgdgd ddgdgg}dddgdgdddgdgdddgdgg}dd d gdgdd d gdgg}t|}t|d vr"|d dddgks|dd ddgkr|St|d vr"|d dddgks|dd ddgkr|St|d vr"|d dd d gks|dd d d gkr|St|d vr"|d dddgks|dd ddgkr|St|d vr"|d dddgks|dd ddgkr|St|gdkrd}nd}td||zdz)aReturns a default Conv2D filter config (list) for a given image shape. Args: shape (Tuple[int]): The input (image) shape, e.g. (84,84,3). Returns: List[list]: The Conv2D filter configuration usable as `conv_filters` inside a model config dict.   @)r=N`T* )rDzThis is the default atari obs shape. You may want to look at one of RLlib's Atari examples for an example of how to wrap an Atari env. z/No default CNN configuration for obs shape {}. aYou can specify `conv_filters` manually through your AlgorithmConfig's model_config. Default configurations are only available for inputs of the following shapes: [42, 42, K], [84, 84, K], [64, 64, K], [10, 10, K]. You may want to use a custom RLModule or a ConnectorV2 for that.)listlenr(r')shape filters_96x96 filters_84x84 filters_64x64 filters_42x42 filters_10x10 atari_helps r.get_filter_configrUs aVQ aVQ r2hM aVQ aVQ r2hM aVQ aVQ aVQ q!fa M aVQ aVQ r2hM aVQ aVQM KKE 5zzVrrr2h!6!6%)Bx:O:O Uv  5!9R#8#8E!""I"brfs6,,,,,,,,,,444444UUUUUUUUUU+/LL 53' (LLLLL^MMXeCM&:;MMMMM`D D D N222222r0