Hui Bingyu

icon indicating a website link https://www.tsinghua.edu.cn/

icon company Tsinghua University icon location Tsinghua University,Hai Dian, Beijing icon location Tsinghua University, Department of Electronic Engineering

Results 3 issues of Hui Bingyu

数据集迁移的问题

3 comment

请问如何实现迁移到自己的数据集上呢？

[BUG/Help] <title>设置多卡参数CUDA_VISIBLE_DEVICES=0,1,2,3 并运行evaluate.sh，报错RuntimeError: handle_0 INTERNAL ASSERT FAILED at "/opt/conda/conda-bld/pytorch_1695392067780/work/c10/cuda/driver_api.cpp":15, please report a bug to PyTorch.

### Is there an existing issue for this? - [X] I have searched the existing issues ### Current Behavior PRE_SEQ_LEN=128 CHECKPOINT=adgen-chatglm-6b-pt-128-2e-2 STEP=3000 CUDA_VISIBLE_DEVICES=0,1,2,3 python3 main.py \ --do_predict \ --validation_file /home/public/dataset/dev.json...

请问本代码能否在单卡4090上做微调呢？最少显存占用多少

1 comment