File size: 413 Bytes
3b5312c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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
|