ResNet18-Pytorch icon indicating copy to clipboard operation
ResNet18-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon samcw
→

Metadata

A model demo which uses ResNet18 as the backbone to do image recognition tasks.Using Pytorch.

Results 1 ResNet18-Pytorch issues
Sort by recently updated
recently updated
newest added

The model lacks a 2d average pooling layer

In model function "forward", after "out = F.avg_pool2d(out, 4)", need do 2d average pooling. Before this, out.size=[-1, 512, 7, 7],after this, out.size=[-1, 512, 1, 1]. define this in net: self.pool2...

CliffNewsted avatar CliffNewsted

← Metadata

53
Stars
14
Forks
Watchers

Owner

verified publisher icon samcw

Metadata

A model demo which uses ResNet18 as the backbone to do image recognition tasks.Using Pytorch.

Back