I'm not quite sure if the StackNN code is fully CUDA compatible, but recently I had some issues trying to use it. Should investigate this and add cuda options everywhere in structs/control_layer.
I'm not quite sure if the StackNN code is fully CUDA compatible, but recently I had some issues trying to use it.
Should investigate this and add cuda options everywhere in structs/control_layer.