gen-efficientnet-pytorch
gen-efficientnet-pytorch copied to clipboard
QAT model fusion problem
how to fuse conv and bn when using this repo to perform quantization aware training?
show in the code snippet,like fuse_model func:
import geffnet model = geffnet.efficientnet_lite0(pretrained=True, drop_rate=0.25, drop_connect_rate=0.2) mode.fuse_model()