IDE和Jupyter Notebook的代码差异:
# IDE获取当前文件的路径
os.path.dirname(__file__)
# ipynb
os.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 tw
import torchvision.models
alexnet_model = torchvision.models.alexnet()
tw.draw_model(alexnet_model, [1, 3, 224, 224])