search

fxmeng / RMNet

RM Operation can equivalently convert ResNet to VGG, which is better for pruning; and can help RepVGG perform better when the depth is large.
210 stars 29 forks source link

inference scrpit #5

Open simplew2011 opened 2 years ago

simplew2011 commented 2 years ago

Can you provide a inference script.

fxmeng commented 2 years ago

My inference script follows this implementation: https://github.com/rwightman/pytorch-image-models/blob/master/benchmark.py