Model Card for Model ID

Summary Length PPO experiment #2
No KL divergence in loss

Model Details

Dataset size: 1024
Epochs: 2
Batch Size: 4 * 8 (using Grad Accu)

Optimizer args: Torch AdamW default, except

LR = 0.0001

Downloads last month: 127

Safetensors

Model size

1.24B params

Tensor type

BF16

Inference Providers NEW

Text Generation

This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.

Model tree for amang1802/Llama3.2-1B-summary-length-exp2

Quantizations

1 model

Collection including amang1802/Llama3.2-1B-summary-length-exp2

PPO experiments

Collection

Using PPO with simpler reward functions • 8 items • Updated 8 days ago