Views
No views yet
1from huggingface_hub import snapshot_download
2from pathlib import Path
3
4model_name = "genbio-ai/GB.MM-Enformer-RNA-1.6B-CDS-ConcatFusion-rna-isoform-expression-ckpt"
5genbio_models_path = Path.home().joinpath('genbio_models', model_name)
6genbio_models_path.mkdir(parents=True, exist_ok=True)
7snapshot_download(repo_id=model_name, local_dir=genbio_models_path)1CONFIG_FILE=... # put the config file path here
2CKPT_PATH=... # put the model checkpoint path here
3
4mgen test --config $CONFIG_FILE \
5 --data.batch_size 16 \
6 --trainer.logger null \
7 --model.strict_loading False \
8 --model.reset_optimizer_states True \
9 --ckpt_path $CKPT_PATH