cherry.nn.init

robotics_init_

robotics_init_(module, gain=None)

[Source]

Description

Default initialization for robotic control.

Credit

Adapted from Ilya Kostrikov's implementation, itself inspired from OpenAI Baslines.

Arguments

Returns

Example

linear = nn.Linear(23, 5)
kostrikov_robotics_(linear)

atari_init_

atari_init_(module, gain=None)

[Source]

Description

Default initialization for Atari environments.

Credit

Adapted from Ilya Kostrikov's implementation, itself inspired from OpenAI Baslines.

Arguments

Returns

Example

linear = nn.Linear(23, 5)
atari_init_(linear)