search

google-research / robotics_transformer

Apache License 2.0
1.29k stars 148 forks source link

The initial values of the network_state for transformer_network #14

Open Zed-Wu opened 1 year ago

Zed-Wu commented 1 year ago

I find that in the unit test (e.g., transformer_network_test.py), the network_state is randomly sampled as:

network_state = tensor_spec.sample_spec_nest(
        network.state_spec, outer_dims=[BATCH_SIZE])

If I want to do inference in my own environment, what are the initial values of the network_state?