From the log, it seems the issue lies in loading your dataset. Try to re-create your dataset. Recommended to try it out with just a small dataset first. Thanks and have a great day ahead!