model_config.mae_autobin.encoder.hidden_dim=768 \
model_config.mae_autobin.decoder.hidden_dim=512 \
model_config.mae_autobin.encoder.depth=12 \
model_config.mae_autobin.decoder.depth=6 \
model_config.mae_autobin.encoder.heads=12 \
model_config.mae_autobin.decoder.heads=8 \
model_config.mae_autobin.bin_num=100 \
dataset_config.rnaseq.bin_num=100 \
dataset_config.rnaseq.bin_type=auto_bin \
training.trainer=lightning \
trainer.params.accelerator=gpu \
trainer.params.devices=1 \
trainer.params.num_nodes=64 \
trainer.params.precision=bf16 \
training.prefetch_factor=200 \
training.num_workers=1 \
training.checkpoint_interval=1000 \
training.batch_size=128 \
scheduler.interval=step \
scheduler.params.warmup_steps=9766 \
scheduler.params.max_lr=1e-4 \
optimizer.params.lr=1e-4 \
optimizer.params.weight_decay=0 \
dataset_config.rnaseq.value_mask_prob=0.3 \
dataset_config.rnaseq.zero_mask_prob=0.03 \
dataset_config.rnaseq.replace_prob=0.8 \
dataset_config.rnaseq.random_token_prob=0.1 \
trainer.params.accumulate_grad_batches=8 \
optimizer.type=adam_w \
trainer.params.gradient_clip_val=1 \