Do you have multiple GPUs? Currently, we don’t support using multiple cards through torch.nn.DataParallel. Please apply the following patch to https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch/blob/master/train.py#L74 13.
# Original version
    net = DataParallel(net).cuda()
# Patched version
    net = net.cuda()