bglick13 commited on
Commit
40c95d2
·
1 Parent(s): d3cb80c

Update value_function/config.json

Browse files
Files changed (1) hide show
  1. value_function/config.json +1 -1
value_function/config.json CHANGED
@@ -1 +1 @@
1
- {"in_channels": 14, "down_block_types": ["DownResnetBlock1D", "DownResnetBlock1D", "DownResnetBlock1D", "DownResnetBlock1D"], "up_block_types": [], "out_block_type": "ValueFunction", "mid_block_type": "ValueFunctionMidBlock1D", "block_out_channels": [32, 64, 128, 256], "layers_per_block": 1, "always_downsample": true, "sample_size": 65536, "out_channels": 14, "extra_in_channels": 0, "time_embedding_type": "positional", "use_timestep_embedding": true, "flip_sin_to_cos": false, "downscale_freq_shift": 1, "norm_num_groups": 8}
 
1
+ {"in_channels": 14, "down_block_types": ["DownResnetBlock1D", "DownResnetBlock1D", "DownResnetBlock1D", "DownResnetBlock1D"], "up_block_types": [], "out_block_type": "ValueFunction", "mid_block_type": "ValueFunctionMidBlock1D", "block_out_channels": [32, 64, 128, 256], "layers_per_block": 1, "always_downsample": true, "sample_size": 65536, "out_channels": 14, "extra_in_channels": 0, "time_embedding_type": "positional", "use_timestep_embedding": true, "flip_sin_to_cos": false, "downscale_freq_shift": 1, "norm_num_groups": 8, "act_fn": "mish"}