prajwalsingh / resnet18-tf2

A simple TensorFlow 2 implementation of ResNet-18

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


Languages

Language:Jupyter Notebook 64.8%Language:Python 35.2%