trace efficientnet的时侯遇到的错,解决方案是set_swish,
net = EfficientNet.from_name('efficientnet-b4')
net.set_swish(memory_efficient=False)
trace_model(net,'Efficientent_b4_best.ckpt')
试了下,成功。
trace efficientnet的时侯遇到的错,解决方案是set_swish,
net = EfficientNet.from_name('efficientnet-b4')
net.set_swish(memory_efficient=False)
trace_model(net,'Efficientent_b4_best.ckpt')
试了下,成功。