Downloads · 30 days
0
smohammadi/torchtune-modelcards-test_20241003183117
torchtune-modelcards-test_20241003183117 is a machine learning model from smohammadi. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is a finetuned version of meta-llama/Llama-3.2-3B-Instruct on the tatsu-lab/alpaca dataset.
Downloads · 30 days
0
Access
Public
Updated Oct 3, 2024
Repo size
—
Likes
0
Public
Click a slice to open those files.
Other1.5 KB · 58%
From the Hugging Face model README
This model is a finetuned version of meta-llama/Llama-3.2-3B-Instruct on the tatsu-lab/alpaca dataset.
More information needed
More information needed
This model was trained using the torchtune library using the following command:
ppo_full_finetune_single_device.py \
--config ./target/7B_full_ppo_low_memory_single_device.yaml \
device=cuda \
metric_logger._component_=torchtune.utils.metric_logging.WandBLogger \
metric_logger.project=torchtune_ppo \
forward_batch_size=2 \
batch_size=64 \
ppo_batch_size=32 \
gradient_accumulation_steps=16 \
compile=True \
optimizer._component_=bitsandbytes.optim.PagedAdamW \
optimizer.lr=3e-4