tracel-ai / models

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

Add ResNet implementation with pre-trained weights #13

Closed laggui closed 8 months ago

laggui commented 9 months ago

Issue #11

This PR adds a Burn implementation for the ResNet family of computer vision models. Implementation and weights are based on torchvision.

TO-DO

laggui commented 9 months ago

Addressed the changes for the current version.

Would be great to have a version that includes the fixes but that also depends on Candle releasing a new version with the module parameters fix for pickle.