IDE和Jupyter Notebook的代码差异:

    1. # IDE获取当前文件的路径
    2. os.path.dirname(__file__)
    3. # ipynb
    4. os.path.abspath("")

    关于参数设置

    1. # ipynb需要下面两个
    2. parser.add_argument("--verbosity", help="increase output verbosity") # ipynb需要这个
    3. args = parser.parse_args(args=[])
    4. # py里只需要下面一句
    5. args = parser.parse_args()

    tensorwatch实例:

    1. import tensorwatch as tw
    2. import torchvision.models
    3. alexnet_model = torchvision.models.alexnet()
    4. tw.draw_model(alexnet_model, [1, 3, 224, 224])