search

NVIDIA-AI-IOT / nanosam

A distilled Segment Anything (SAM) model capable of running real-time with NVIDIA TensorRT
Apache License 2.0
616 stars 52 forks source link

Can you provide the torch model of ResNet18? #4

Open superAiken opened 11 months ago

superAiken commented 11 months ago

Thanks for your great work! I cannot convert the ONNX model provided by you into a Torch model. Can you provide the torch model of ResNet18? Thanks again for this fantastic job.

shawnricecake commented 8 months ago

any update for the pytorch model?

CarryHJR commented 7 months ago

i follow the readme to train res18 using coco dataset with a 4090 GPU, about 4 days, , you guys can try my checkpoints https://drive.google.com/file/d/1D2zSaCmhJf-ANymRCVX-Xtdr44upAL6r