一、場景描述
我們希望檢測一類條件,如果條件不滿足就中斷程序運行,並且可以給出錯誤信息描述。一類簡單的解決方案就是採用斷言。
二、Demo
assert condition, "error info"
例如:
assert os.path.isfile(args.resume), 'Error: no checkpoint directory found!'
# 斷言的作用
我們希望檢測一類條件,如果條件不滿足就中斷程序運行,並且可以給出錯誤信息描述。一類簡單的解決方案就是採用斷言。
assert condition, "error info"
例如:
assert os.path.isfile(args.resume), 'Error: no checkpoint directory found!'
# 斷言的作用
報錯 原因分析 涉及到梯度傳播的tensor變量不能夠直接numpy()轉numpy,需要使用: var.detach().numpy()