bug描述 Describe the Bug Bug: ImportError: cannot import name 'MSRA' from 'paddle.fluid.initializer' (/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/initializer.py) 运行代码: import os import paddleslim.quant a...
🚀 Feature We would like an option in torch.onnx.export which turns off Conv+BN fusion. Motivation We are using ONNX for training. When our trainer starts, it exports a pytorch model in both eval mode and training mode and then compile th...
import torch import torch.nn as nn class SimpleCNN(nn.Module): def __init__(self): super(SimpleCNN, self).__init__() self.conv1 = nn.Conv2d(1, 20, 5) self.pool = nn.MaxPool2d(2, 2) self.bn1 = nn.BatchNorm2d(20) self.conv2 = nn.Conv2d(20, 50, 5) self.bn2 = nn...
conv.py -> build/lib.linux-x86_64-3.8/apex/pyprof/prof copying apex/pyprof/prof/data.py -> build/lib.linux-x86_64-3.8/apex/pyprof/prof copying apex/pyprof/prof/pooling.py -> build/lib.linux-x86_64-3.8/apex/pyprof/prof copying apex/pyprof/prof/activation.py -> build/lib.linux-...