I am getting following error when I try to optimize the network with kFLOAT precision. I have tried increasing workspace size to maximum value of size_t (1<<31) but still got the same error. Can anybody tell me how I can fix this?
ERROR: Internal error: could not find any implementation for node conv1, try increasing the workspace size with IBuilder::setMaxWorkspaceSize() ERROR: cudnnBuilder2.cpp (452) - OutOfMemory Error in buildSingleLayer python: ctp.cpp:71: void caffeToGIEModel(const char*, const char*, const char*, uint16_t): Assertion `engine' failed. Aborted (core dumped)