tracel-ai / models

Models and examples built with Burn
Apache License 2.0
169 stars 23 forks source link

Add ResNet implementation with pre-trained weights #13

Closed laggui closed 7 months ago

laggui commented 8 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 7 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.