IDE和Jupyter Notebook的代码差异:
# IDE获取当前文件的路径os.path.dirname(__file__)# ipynbos.path.abspath("")
关于参数设置
# ipynb需要下面两个parser.add_argument("--verbosity", help="increase output verbosity") # ipynb需要这个args = parser.parse_args(args=[])# py里只需要下面一句args = parser.parse_args()
tensorwatch实例:
import tensorwatch as twimport torchvision.modelsalexnet_model = torchvision.models.alexnet()tw.draw_model(alexnet_model, [1, 3, 224, 224])
