Closed relh closed 5 years ago
Okay, well I was able to make some small modifications and train successfully on my GPU without those things. I'll put up a PR to show what I did.
EDIT: I changed a lot and made the whole thing run in a multiprocess/distributed fashion and it's a lot to disentangle it. Gonna bail on the PR but ping me if you find this on Google and need help.
move fast and break things? ;)
EDIT: Thanks for the help though!
What is needed to implement GPUs?