PornX_AI / training_config.yml
ChromiumPlutoniumAI's picture
Create training_config.yml
f9583c8 verified
raw
history blame contribute delete
131 Bytes
training:
batch_size: 4
epochs: 3
learning_rate: 2e-5
gradient_accumulation: 8
mixed_precision: "fp16"
save_steps: 500