Skip to content

jimmyyhwu/resnet18-tf2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 

Repository files navigation

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