Hi:
I tried to test the performance of autotvm for a Resnet_v2_50 on a Nvidia Tesla T4 card.
At first, I just use the default log file for the relay.build
function to build the graph, lib, params
, and when I run it with the module created by graph_runtime.create
, it works fine.
But after I tune the network on the T4 card, get the log file, build the graph, lib, params
with the relay.build
. When I run the module created by graph_runtime.create
, it display
TVMError: Check failed: ret == 0 (-1 vs. 0) : Assert fail: (58 == int32(arg2.shape[2])), Argument arg2.shape[2] has an unsatisfied constraint
The only difference is the log file used to run relay.build
. So any suggestions? Thanks