3 ms·
So to catch up, AID2 is leveraging a cute offering by Google Colab which will provide any Google account a few hours of free access to a fairly powerful GPU. T
by zaroth 7y ago
So to catch up, AID2 is leveraging a cute offering by Google Colab which will provide any Google account a few hours of free access to a fairly powerful GPU.
To use Colab under this arrangement, each person playing the game has to load the model into their personal VM, which is being billed by Google as 6GB of data transfer from GCP into Colab.
The obvious questions;
1) Was there a place they could have hosted the model closer to Colab so that they weren’t being charged for egress bandwidth — and also, ideally, so that 6GB of data wasn’t actually being moved very far?
2) The underlying model is 6GB, but I’m curious how much memory is required for an individual user’s world state and how hard it would be to have a single GPU handling multiple user sessions?
Presumably it would be possible to multiplex multiple sessions with a single GPU? You would have to serialize the game state, receive the next input, load the prior state, feed the new input through the model, return the resulting text output, and re-serialize the state until the next input comes through.
What I don’t know if that’s at all practical based on the amount of data that would have to be serialized? Is the 6GB model data separate and static throughout the game, with an isolated block of data for the current world-state? Or does playing the game fundamentally alter the state of the model, meaning you would have to reload the whole thing just to process the next command?
- Akababa 7y agoGoogle Drive is actually perfect for this as it lets you mount the drive (presumably as NAS) so you can read from it directly. I set up a notebook for it but unfortunately the shared folder got rate-limited. It still works if you make a copy of the files in your own Google drive though. https://colab.research.google.com/github/Akababa/AIDungeon/blob/patch-2/AIDungeon_2.ipynb https://colab.research.google.com/github/Akababa/AIDungeon/b...
- abecedarius 7y agoThe obvious question to me is why must the cost fall on the provider instead of the player? If micropayments were easy enough then this would've been sustainable.
- killerstorm 7y ago"World state" is the text people see on script. It can reside in the browser. So a server can be completely stateless, i.e. it receives text so far (say, 5 KB), applies GPT-2 generate and returns. The problem is that GPT-2 generate seems to be very computationally intensive. As I understand, it actually does number crunching with all these 6 GB of data, so it takes 5-10 seconds even on GPU (K80, at least, is that slow). Is GPU capable of running multiple GPT-2 generate in parallel? No idea. Assuming that a high-end GPU would be able to produce response in 2 seconds, you can only run maybe 10 concurrent sessions per server if you want fast response time.