Skip to content

Expose scGPT fine-tuning epochs and batch size - #118

Open
Daesun-Ryu wants to merge 1 commit into
openproblems-bio:mainfrom
Daesun-Ryu:fix/111-scgpt-training-params
Open

Daesun-Ryu wants to merge 1 commit into
openproblems-bio:mainfrom
Daesun-Ryu:fix/111-scgpt-training-params

Conversation

@Daesun-Ryu

Copy link
Copy Markdown

Summary

  • Expose --epochs and --batch_size, preserving production defaults of 15 and 64 (minimum 1).
  • Wire these values to training, validation, and embedding.
  • Add Viash info.test_default (epochs 2, batch size 4) without changing normal benchmark settings.
  • Add eight CPU-only regression tests and a CI job.

Validation

  • Production and Viash-injected regression suites: 8/8 each.
  • Viash 0.9.4: merged config, injection, Docker-engine executable build without setup, and CLI help.
  • Earlier Phase 3 native boundary probe: 13/13 (validation-only; ML execution intercepted).

Limits

  • No real Docker/ML integration, GPU/T4 OOM or VRAM measurement, or biological benchmark-quality validation.
  • GitHub-hosted CI needs review after PR submission.

Refs #111

@rcannood rcannood left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@Daesun-Ryu Thanks for the PR; exposing the epochs and batch size makes sense.

Could you review my comments and apply the changes?

Could you also add yourself as a contributor to _viash.yaml?

Comment on lines +12 to +23

scgpt_training_parameters:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.12"
- run: python -m pip install PyYAML==6.0.3
- run: >-
python -B -m unittest discover
-s src/methods/scgpt_finetuned -p test_training_params.py -v

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Undo these changes to our CI

Suggested change
scgpt_training_parameters:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: actions/setup-python@v5
with:
python-version: "3.12"
- run: python -m pip install PyYAML==6.0.3
- run: >-
python -B -m unittest discover
-s src/methods/scgpt_finetuned -p test_training_params.py -v

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Remove this file

Comment thread CHANGELOG.md
Comment on lines +52 to +53
* Expose `methods/scgpt_finetuned`'s epochs and batch size as arguments, preserving defaults of 15 and 64.
Batch size also applies to validation and embedding (issue #111).

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This isn't a bug fix but a minor change.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants