Whether to annotate out pytorch error checking code?

ok. Thanks for your reply.