Adding ZERO support

#1
by taesiri - opened

Hey there!

This tool is really cool, I modified the code to use the GPU Zero which makes the process slightly faster.

PyTorch Image Models org

@taesiri cool, yeah, thought about that ... for me at least the biggest delay for occasional use is the weight download. If you're sitting down with it using the same models with different images this would have a big impact though.

Sign up or log in to comment