| codebook_config: | |
| freeze: true | |
| num_codes: 8192 | |
| embed_dim: 13 | |
| entropy_loss_weight: 0.1 | |
| commitment_loss_weight: 0.25 | |
| encoder_config: | |
| freeze: true | |
| decoder_config: | |
| input_dim: 128 | |
| trunk: | |
| num_blocks: 4 | |
| sequence_state_dim: 128 | |
| pairwise_state_dim: 32 | |
| max_recycles: 1 | |
| structure_module: | |
| c_s: 128 | |
| c_z: 32 | |
| no_blocks: 8 | |
| no_angles: 7 | |
| gradient_checkpointing: false | |