VAE训练数据集选择的问题

您好，请问我拿MVTec格式的数据集训练VAE时，数据集下有好多类物体，每个物体都有一个训练集，请问在配置文件中定义数据集路径的地方，我是随意选择一个物体的训练集训练VAE就可以了嘛