_cdata) return x # 被替换OP 原OP 自定义OP F.conv2d = Hook(F.conv2d,_conv2d) ... F.batch_norm = Hook(F.batch_norm,_batch_norm) F.relu = Hook(F.relu,_relu) ...
確定! 回上一頁