tracel-ai / models

Models and examples built with Burn
Apache License 2.0
180 stars 24 forks source link

Add ResNet implementation #11

Closed laggui closed 8 months ago

laggui commented 9 months ago

ResNet-18 (and other variants) are often the first baseline in image classification tasks.

We should add an implementation that could serve as an example for architecture implementations with Module and different built-in modules.