6 ms·
NVIDIA just released the code: https://github.com/nvlabs/spade/ https://github.com/nvlabs/spade/
by AdamDKing 7y ago
NVIDIA just released the code: https://github.com/nvlabs/spade/ https://github.com/nvlabs/spade/
- bitL 7y agoWhat software do you use to make those 3D DNN architecture images?
- knolan 7y agoI’ve no idea what nvidia use but you could do this pretty easily in Blender with the Freestyle NPR renderer and the built in import images as planes add on.
- AdamDKing 7y agoFor all the 3D diagrams that I made (including the animated one at the end) I wrote code that used https://threejs.org/ https://threejs.org/ and my custom library. It worked, but with a lot of hassle. In the future I'll likely try using Blender.
- scoutt 7y ago"To reproduce the results reported in the paper, you would need an NVIDIA DGX1 machine with 8 V100 GPUs."
- AdamDKing 7y agoThat line refers to training the model from scratch. You can still run the trained model very quickly with one "cheap" GPU. That said, I'm not sure why one wouldn't get a similar result training on the EC2 or GCE instances that have 8 V100s. Or even training with fewer GPUs but accumulating gradients to get the same batch size.