colab中报错no module named ‘drive‘解决

参考：https://zhuanlan.zhihu.com/p/105647700
侵删，原文中可解决的问题更多。

在调用其他.py文件之前加

import sys
sys.path.append('/content')

其中引号中内容可改。

以下 例一与例二等同。
例一：

import sys
sys.path.append('/content')

from drive.MyDrive.malware.code.UTransformer.bert import BatchGeneratorForBERT

例二：

import sys
sys.path.append('/content/drive/MyDrive/malware/code')

from UTransformer.bert import BatchGeneratorForBERT