Quote Originally Posted by djtayz View Post
There already is a Colab for it (the link is from the github page)

https://colab.research.google.com/dr...GH?usp=sharing
Ok yeah I had a look at that earlier yesterday but I didn't see where you can change model to model B or A, I'll give it another look though. Thanks for the reply btw..