Some of the graph can’t be optimized (they were showing N/A), resnet18 is okay and works fine.
Here are my code to optimize with tensorcore.
Hi @twmht , I tried executing your script and I get the following error :
I traced the error to its source and it seems to be from the ToMixedPrecision() Pass. Has something to do with float16 handling by the metascheduler. Can you confirm if you have faced an error similar to this?
I also see this issue here but it has been stagnant for a while now.
Can you please let me know if you were able to find a workaround for this? TIA and have a nice day!
Regards,
Krishna