Skip to content

Latest commit

 

History

1 Commit

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 

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

About

A simple TensorFlow 2 implementation of ResNet-18

Topics

Resources

Stars

29 stars

Watchers

1 watching

Forks

Contributors

Languages