我正在尝试用我自己的数据训练pixelda（https://github.com/tensorflow/models/tree/master/research/domain_adaptation）。我已经有.tfrecords了。
当我从readme.md运行bazel run domain_adaptation/pixel_domain_adaptation:pixelda_train -- --dataset_dir $DSN_DATA_DIR --source_dataset mnist --target_dataset mnist_m命令并插入我自己的数据作为参数时，我得到ValueError: Name of dataset unknown xy。
经过一些研究后，我发现脚本只通过dataset_factory脚本接受mnist数据。 如何调整代码以便我可以在网络上应用自己的数据？