COCOWhole Body Training

[berkearda]

I'm trying to train the AlphaPose on COCOWhole Body dataset. It can train it for 2 successive epoch but before third epoch in validation stage I get an error as followed: Traceback (most recent call last): File "/content/AlphaPose/scripts/train.py", line 345, in main() File "/content/AlphaPose/scripts/train.py", line 302, in main gt_AP = validate_gt(m.module, opt, cfg, heatmap_to_coord) File "/content/AlphaPose/scripts/train.py", line 243, in validate_gt res = evaluate_mAP(os.path.join(opt.work_dir, 'test_gt_kpt.json'), ann_type='keypoints', ann_file=os.path.join(cfg.DATASET.VAL.ROOT, cfg.DATASET.VAL.ANN), halpe=halpe) File "/content/AlphaPose/alphapose/utils/metrics.py", line 102, in evaluate_mAP cocoEval.evaluate() File "/root/.local/lib/python3.10/site-packages/halpecocotools-0.0.0-py3.10-linux-x86_64.egg/halpecocotools/cocoeval.py", line 183, in evaluate self.ious = {(imgId, catId): computeIoU(imgId, catId) \ File "/root/.local/lib/python3.10/site-packages/halpecocotools-0.0.0-py3.10-linux-x86_64.egg/halpecocotools/cocoeval.py", line 183, in self.ious = {(imgId, catId): computeIoU(imgId, catId) \ File "/root/.local/lib/python3.10/site-packages/halpecocotools-0.0.0-py3.10-linux-x86_64.egg/halpecocotools/cocoeval.py", line 269, in computeOks sigmas = sigmas[0 : body_kpnum] UnboundLocalError: local variable 'sigmas' referenced before assignment Exception ignored in: <alphapose.utils.metrics.evaluate_mAP..NullWriter object at 0x7fad235d6ef0> AttributeError: 'NullWriter' object has no attribute 'flush'