deepinsight / deepinsight/insightface

Failed to allocate CPU Memory: When trying to run train.py under recognition folder

Open
#957 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
29.7k
Forks
6.1k
PR merge metrics
No merged PRs in 30d

Description

CUDA_VISIBLE_DEVICES='0,1,2,3' python -u train.py --network r100 --loss arcface --dataset emore
gpu num: 4
prefix ./models/r100-arcface-emore/model
image_size [112, 112]
num_classes 85742
Called with argument: Namespace(batch_size=256, ckpt=3, ctx_num=4, dataset='emore', frequent=20, image_channel=3, kvstore='device', loss='arcface', lr=0.1, lr_steps='100000,160000,220000', models_root='./models', mom=0.9, network='r100', per_batch_size=64, pretrained='', pretrained_epoch=1, rescale_threshold=0, verbose=2000, wd=0.0005) {'loss_m1': 1.0, 'loss_m2': 0.5, 'loss_m3': 0.0, 'net_act': 'prelu', 'emb_size': 512, 'data_rand_mirror': True, 'num_layers': 100, 'loss_name': 'margin_softmax', 'val_targets': ['lfw', 'cfp_fp', 'agedb_30'], 'ce_loss': True, 'net_input': 1, 'image_shape': [112, 112, 3], 'net_blocks': [1, 4, 6, 2], 'fc7_lr_mult': 1.0, 'ckpt_embedding': True, 'net_unit': 3, 'net_output': 'E', 'count_flops': True, 'num_workers': 1, 'batch_size': 256, 'memonger': False, 'data_images_filter': 0, 'dataset': 'emore', 'num_classes': 85742, 'fc7_no_bias': False, 'loss': 'arcface', 'data_color': 0, 'loss_s': 64.0, 'dataset_path': '../datasets/faces_emore', 'data_cutoff': False, 'net_se': 0, 'net_multiplier': 1.0, 'fc7_wd_mult': 1.0, 'network': 'r100', 'per_batch_size': 64, 'net_name': 'fresnet', 'workspace': 256, 'max_steps': 0, 'bn_mom': 0.9}
0 1 E 3 prelu False
Network FLOPs: 24.2G
INFO:root:loading recordio ../datasets/faces_emore/train.rec...
header0 label [5822654. 5908396.]
id2range 85742
5822653
rand_mirror True
../datasets/faces_emore/lfw.bin
Image isze
[112, 112]
loading bin 0
loading bin 1000
loading bin 2000
loading bin 3000
loading bin 4000
loading bin 5000
loading bin 6000
loading bin 7000
loading bin 8000
loading bin 9000
loading bin 10000
loading bin 11000
(12000, 3, 112, 112)
ver lfw
../datasets/faces_emore/cfp_fp.bin
Image isze
[112, 112]
Traceback (most recent call last):
File "train.py", line 389, in
main()
File "train.py", line 385, in main
train_net(args)
File "train.py", line 276, in train_net
data_set = verification.load_bin(path, image_size)
File "eval/verification.py", line 197, in load_bin
data = nd.empty((len(issame_list) * 2, 3, image_size[0], image_size[1]))
File "/home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/ndarray/utils.py", line 103, in empty
return _empty_ndarray(shape, ctx, dtype)
File "/home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/ndarray/ndarray.py", line 3955, in empty
return NDArray(handle=_new_alloc_handle(shape, ctx, False, dtype))
File "/home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/ndarray/ndarray.py", line 141, in _new_alloc_handle
ctypes.byref(hdl)))
File "/home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/base.py", line 253, in check_call
raise MXNetError(py_str(_LIB.MXGetLastError()))
mxnet.base.MXNetError: [15:17:21] src/storage/./cpu_device_storage.h:75: Failed to allocate CPU Memory
Stack trace:
[bt] (0) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/libmxnet.so(+0x4b03ab) [0x7f4ad02313ab]
[bt] (1) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/libmxnet.so(+0x2e618c1) [0x7f4ad2be28c1]
[bt] (2) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/libmxnet.so(+0x2e6a96f) [0x7f4ad2beb96f]
[bt] (3) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/libmxnet.so(mxnet::NDArray::NDArray(mxnet::TShape const&, mxnet::Context, bool, int)+0x5d0) [0x7f4ad22b25b0]
[bt] (4) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/site-packages/mxnet/libmxnet.so(MXNDArrayCreateEx+0x22d) [0x7f4ad22b307d]
[bt] (5) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/lib-dynload/_ctypes.so(ffi_call_unix64+0x4c) [0x7f4b04f1857c]
[bt] (6) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/lib-dynload/_ctypes.so(ffi_call+0x1f5) [0x7f4b04f17cd5]
[bt] (7) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/lib-dynload/_ctypes.so(_ctypes_callproc+0x3e6) [0x7f4b04f0f376]
[bt] (8) /home/testuser/anaconda2/envs/pytorch/lib/python2.7/lib-dynload/_ctypes.so(+0x9db3) [0x7f4b04f06db3]

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.