-
Notifications
You must be signed in to change notification settings - Fork 101
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
训练出错(https://github.com/MegEngine/Models/tree/master/official/quantization) #63
Comments
aspnetcs
changed the title
训练出错
训练出错(https://github.com/MegEngine/Models/tree/master/official/quantization)
Sep 5, 2020
ImageNet.zip |
imagenet,过于庞大,可用于训练测试的有小型数据集吗? |
首先,models里面的模型是用于复现,用小数据集如mnist的确可以,但是没有人会在mnist上做试验,如果你想要玩一些toy模型,自己写一个dataloader就可以 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
执行 python3 train.py -a resnet18 -d /home/rootx/Models/official/quantization/dataset/flowers-recognition/ --mode normal
出错了,缺少文件,咋办?
root@rootx-virtual-machine:/home/rootx/Models/official/quantization# python3 train.py -a resnet18 -d /home/rootx/Models/official/quantization/dataset/flowers-recognition/ --mode normal
err: Failed to load CUDA Driver API library
err: failed to load cuda func: cuCtxGetCurrent
err: failed to load cuda func: cuCtxGetCurrent
05 09:43:15[mgb] ERR cudaGetDeviceCount failed: CUDA driver version is insufficient for CUDA runtime version (err 35)
05 09:43:15[mgb] WRN cuda unavailable: CUDA driver version is insufficient for CUDA runtime version(35) ndev=-1
05 09:43:16 preparing dataset..
05 09:43:16 WRN devkit directory /home/rootx/Models/official/quantization/dataset/flowers-recognition/ILSVRC2012_devkit_t12 does not exists
05 09:43:16 checksum devkit tar file /home/rootx/Models/official/quantization/dataset/flowers-recognition/ILSVRC2012_devkit_t12.tar.gz ...
Traceback (most recent call last):
File "train.py", line 309, in
main()
File "train.py", line 65, in main
worker(0, 1, args)
File "train.py", line 152, in worker
train_dataset = data.dataset.ImageNet(args.data, train=True)
File "/usr/local/lib/python3.6/dist-packages/megengine/data/dataset/vision/imagenet.py", line 97, in init
self._prepare_devkit()
File "/usr/local/lib/python3.6/dist-packages/megengine/data/dataset/vision/imagenet.py", line 245, in _prepare_devkit
calculate_md5(raw_file) == checksum
File "/usr/local/lib/python3.6/dist-packages/megengine/data/dataset/vision/utils.py", line 56, in calculate_md5
with open(filename, "rb") as f:
FileNotFoundError: [Errno 2] No such file or directory: '/home/rootx/Models/official/quantization/dataset/flowers-recognition/ILSVRC2012_devkit_t12.tar.gz'
root@rootx-virtual-machine:/home/rootx/Models/official/quantization#
环境
1.系统环境:
2.MegEngine版本:
3.python版本:
4.模型名称:
复现步骤
请提供关键的代码片段便于追查问题
请提供完整的日志及报错信息
The text was updated successfully, but these errors were encountered: