search

coreylowman / dfdx

Deep learning in Rust, with shape checked tensors and neural networks
Other
1.69k stars 98 forks source link

Add integration test for mobile net #773

Open coreylowman opened 1 year ago

coreylowman commented 1 year ago

With #767 now added, we can integration test against more complex image networks!

This new test should mirror the existing resnet18 integration test, but with a mobile net structure.

Awpteamoose commented 1 year ago

I'd like to contribute this, I'm at the it's ugly but works stage