/resnet18-tf2

A simple TensorFlow 2 implementation of ResNet-18

Primary LanguageJupyter Notebook

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