ImagePromptHelper-gemma3-270M

ImagePromptHelper-gemma3-270M

This model is a fine-tuned version of google/gemma-3-270m on the ImagePromptHelper-v02 (CC BY 4.0) dataset. It achieves the following results on the evaluation set:

• Loss: 0.2502

Model description

This model expands short image prompts into long image prompts. The moun optimizer was used to train this model to see what would happen. The result is much better than my previous attempts.

Intended uses & limitations

This model is intended to be used for image prompt expansion in a variety of ways as determined by the dataset that was used to train it. It is not intended to be used for any other purpose.

Training and evaluation data

I used the moun optimizer to train this model. Here is the LLama Factory config:

<details> <summary>LLama Factory config</summary>

### model
model_name_or_path: google/gemma-3-270m

### method
stage: sft
do_train: true
finetuning_type: full
use_muon: true
seed: 101

### dataset
dataset: image_prompter_v2
template: gemma
cutoff_len: 2048
overwrite_cache: false
preprocessing_num_workers: 12

### output
output_dir: Gemma3/270M/full/image_prompter
logging_steps: 1
save_steps: 2500
save_strategy: steps
plot_loss: true
overwrite_output_dir: false

### train
per_device_train_batch_size: 1
gradient_accumulation_steps: 8
learning_rate: 1.0e-04
num_train_epochs: 2.0
weight_decay: 0.01
adam_beta1: 0.90
adam_beta2: 0.98
max_grad_norm: 1.0
lr_scheduler_type: cosine
warmup_ratio: 0.075
bf16: true

### eval
val_size: 0.01
per_device_eval_batch_size: 1
eval_strategy: steps
eval_steps: 2500

</details>

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 0.0001
• train_batch_size: 1
• eval_batch_size: 1
• seed: 101
• gradient_accumulation_steps: 8
• total_train_batch_size: 8
• optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.98) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
• lr_scheduler_type: cosine
• lr_scheduler_warmup_ratio: 0.075
• num_epochs: 2.0

Training results

Framework versions

• Transformers 4.52.4
• Pytorch 2.7.0+cu128
• Datasets 3.6.0
• Tokenizers 0.21.1