Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 669 Bytes

README.md

File metadata and controls

13 lines (9 loc) · 669 Bytes

resnet18-tf2

The official TensorFlow ResNet implementation does not appear to include ResNet-18 or ResNet-34.

This codebase provides a simple (70 line) TensorFlow 2 implementation of ResNet-18 and ResNet-34, directly translated from PyTorch's torchvision implementation. The model outputs have been verified to match those of the torchvision models with floating point accuracy.

This code was tested with the following package versions:

tensorflow==2.4.1
pytorch==1.2.0
torchvision==0.4.0