train AI model for VFX