Finetune 风格模型

接下来我们尝试在预训练模型之上进行 finetune（微调），训练一个能出指定风格的模型。

1. 准备训练素材

我们预先准备了一组图片（共20张）存储在EC2中。

连接到EC2，将图片上传到 S3 存储桶（不含 jp-style 文件夹），推荐上传到如下路径：(注意修改下面命令中 S3桶名为自己的桶，替换REGION为部署区域，account id 即可)

s3://sagemaker-{REGION}-{YOUR_ACCOUNT_ID}/dataset/

1
2
cd images/jp-style/ aws s3 cp . s3://sagemaker-{REGION}-{YOUR_ACCOUNT_ID}/dataset/ --recursive 

2. 创建训练任务

我们通过 WebUI创建训练任务，首先在训练页下面找到训练 Dreambooth，然后在模型下创建新模型，填入一个名称，源检查点处刷新并选择一个模型。然后切换到参数。（注意不要点最下面的训练模型）

在参数页下，选择实例类型，建议 ml.g4dn.2xlarge, Concepts S3 位置填写1. 准备训练素材步骤中我们上传到的 S3 路径。然后切换到概念页。（注意不要点最下面的训练模型）

在概念页，在数据集目录处填写/opt/ml/input/data/concepts/，在 Filewords 处可以分别填写photo of jpstyle girl 和 photo of girl。其中 jpstyle 可以替换为任意非正常英文单词，可以是字母和数字的组合，如 asdf1234。

最后在最下面，点击训练，看到左侧显示训练任务提交成功。

该模型训练预计需要1小时，我们可以在 All-In-One AI控制台界面训练任务tab下查看。（由于实验账号限制，此步骤无法在Event Engine账号中进行）。

于此同时，您可以继续在 WebUI中探索文生图，或者图生图等。