Open rbunn80110 opened 4 years ago
cool project. I rewrote it in pytorch in order to get training to work. You should provide a train.py file since it isn't obvious how you would train a model like this in keras.
Thanks for your comment and sorry for the late reply. I will add the training script in the project.
cool project. I rewrote it in pytorch in order to get training to work. You should provide a train.py file since it isn't obvious how you would train a model like this in keras.