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)), Argument arg2.shape has an unsatisfied constraint
The only difference is the log file used to run
relay.build. So any suggestions? Thanks