请教一些问题
wenzitracyliu opened this issue · comments
wenzitracyliu commented
修改一下问题:
1、您好,我在运行Run Run Pre-defined problems.ipynb这个文件的时候,导入自己的模型,然后设置batchsize为1,multiprocess为False,单个任务,会占满整个显存。
其实我尝试各种batchsize大小都不行,都是这样,不知道问题出在哪。另外我是这么修改您的代码
除了pip安装,我也试过直接下载你的代码直接运行,也是这个问题。另外我想问问,除了batchsize、multiprocess,还可以调整哪些参数可以降低显存的使用
2、另外就是调用train_bert_multitask会出现eval_feature_desc.json未创建。
最后查到train_and_evaluate(estimator, train_spec, eval_spec)中有误