main
add dataset and edit train.py
add dataset
some edits after first try
Initial commit
update README.md
Update train.sh