So i've been trying to implement Enumerated Shapes and Ranges to try to get away from having to have a crap ton of the same model for various sizes.. and wow its SO SLOW.
Not sure if you've played around with it yet but either i'm doing something wrong or it's just really bad.
Not just the first run but even repeat generations...
Tested with C+N ORIG @ 512x512 (mainly because i forgot to swap to C+G before i rolled back my code)...
Git Main: 1s/it
Stock Pytorch: 1.27s/it
Enumerated: 5.05s/it. (the ranged i forgot to write down but it was similar).
So i've been trying to implement Enumerated Shapes and Ranges to try to get away from having to have a crap ton of the same model for various sizes.. and wow its SO SLOW.
Not sure if you've played around with it yet but either i'm doing something wrong or it's just really bad.
Not just the first run but even repeat generations...
Tested with C+N ORIG @ 512x512 (mainly because i forgot to swap to C+G before i rolled back my code)... Git Main: 1s/it Stock Pytorch: 1.27s/it Enumerated: 5.05s/it. (the ranged i forgot to write down but it was similar).
GIT Main Model:
Enumerated Model: