revert test train geneformer params

Author: nvmvle

sub-packages/bionemo-geneformer/tests/bionemo/geneformer/scripts/test_train_geneformer.py

@@ -69,16 +69,16 @@ def test_main_runs(tmpdir, create_checkpoint_callback: bool, data_path: Path):
             result_dir=result_dir,
             wandb_project=None,
             wandb_offline=True,
-            num_steps=5,  # Run for more steps to ensure metrics are logged
+            num_steps=5,
             limit_val_batches=1,
             val_check_interval=2,
             num_dataset_workers=0,
             biobert_spec_option=BiobertSpecOption.bert_layer_local_spec,
             lr=1e-4,
             micro_batch_size=2,
             accumulate_grad_batches=1,
-            cosine_rampup_frac=0.1,
-            cosine_hold_frac=0.1,
+            cosine_rampup_frac=0.01,
+            cosine_hold_frac=0.01,
             precision="bf16-mixed",
             experiment_name="test_experiment",
             resume_if_exists=False,