R1 / config.yaml
hackergeek98's picture
Create config.yaml
bd57c6c verified
raw
history blame contribute delete
286 Bytes
project_name: phi4-training
model: microsoft/phi-4 # Replace with "microsoft/phi-4" if available
data_path: ./
task: text-generation # Or your task (e.g., text-classification)
learning_rate: 2e-5
batch_size: 2
epochs: 3
fp16: true
push_to_hub: true
repo_id: "hackergeek98/phi4-trained"