U ʹj1@s8ddlZddlZd ddZd ddZddZd d ZdS) NFc CsN|js |jrtdt|}t|j|j|j|j|j |j|j|\|_|_|SNzFusion only for eval!) trainingAssertionErrorcopydeepcopyfuse_conv_bn_weightsweightbias running_mean running_vareps)convbn transposeZ fused_convrT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/nn/utils/fusion.pyfuse_conv_bn_evals   rcCs|j}|dk r|jn|} |dkr*t|}|dkrs