-
-
Notifications
You must be signed in to change notification settings - Fork 65
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add pytorch/torchvision porting and comparison script #225
Conversation
198f510
to
91d75cf
Compare
This script and an older version that @darsnack had posted as a gist both work, but I find that |
Here is also a script for porting weights. |
Yep, that's the one I mentioned. Here too |
Can you take a quick allocation profile with and show the flamegraph here using PProf or the VSCode profile view? |
91d75cf
to
658588f
Compare
Ah, @ToucheSir, I neglected to mention something which on a re-run I find might be the cause of this. I was using @darsnack's script to port weights but I added an |
No description provided.