Hitdahit / ResNet_LibTorch

A small example of using new PyTorch C++ frontend to implement ResNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ResNet_LibTorch

A small example of using new PyTorch C++ frontend to implement ResNet.

Inspired by the example suggested by the official PyTorch c++ doc, I spent a weekend writing a small example of ResNet.

You may need to:

  • Update your CMake to the lastest (Mine is 3.12). I started with version 3.5 and it could not work.

About

A small example of using new PyTorch C++ frontend to implement ResNet


Languages

Language:C++ 96.6%Language:CMake 3.4%