PyTorch CUDA runtime error (59) : device-side assert triggered

出現這個問題緣由不少,尤爲是使用Jupyter時,通常而言,都是代碼出錯,或者屢次運行。檢查輸入和模型是否批評。例如json 在運行BertModel時,若是num_labels指定錯了,和你輸入不匹配,就會形成這種問題。it   bert_config = BertConfig(vocab_size_or_config_json_file=30522,
相關文章
相關標籤/搜索