- Python
- SimpleITK
- NumPy
conda install numpy -y
everyoneelse
everyoneelse
/ grpo_demo.py
Created
February 2, 2025 08:02
— forked from willccbb/grpo_demo.py
GRPO Llama-1B
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # train_grpo.py | |
| import re | |
| import torch | |
| from datasets import load_dataset, Dataset | |
| from transformers import AutoTokenizer, AutoModelForCausalLM | |
| from peft import LoraConfig | |
| from trl import GRPOConfig, GRPOTrainer | |
| # Load and prep dataset |
everyoneelse
/ README.md
Created
January 30, 2024 05:31
— forked from fepegar/README.md
Python function to create maximum intensity projections using SimpleITK