Can any deep learning model be used in nimble, such as yolov5??

[umairjavaid]

No description provided.

[gyeongin]

I am not familiar with the YOLO v5 model, so let me list the general requirements of Nimble.
Nimble requires a neural network module to have the following properties:

• CUDA only (should not entail any CPU operations)
• Static (performs the same set of computations for different runs; e.g., the shape of the input image should be fixed)

[bgchun]

We can extend Nimble to relax the above requirements: handling CPU operations and dynamic shapes. However, these extensions are not part of the open-sourced Nimble.

[umairjavaid]

Thank you so much!