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:

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

Base model

google/gemma-3-270m

(147)