I have this error when run train.py:
Traceback (most recent call last):
File "H:/PyCharm project/retinanet/train.py", line 101, in
train(epoch)
File "H:/PyCharm project/retinanet/train.py", line 55, in train
for batch_idx, (inputs, loc_targets, cls_targets) in enumerate(trainloader):
File "D:\Python\lib\site-packages\torch\utils\data\dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "D:\Python\lib\site-packages\torch\utils\data\dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
File "D:\Python\lib\site-packages\torch\utils\data_utils\worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "H:\PyCharm project\retinanet\utils\util.py", line 28, in collate_fn
loc_target, cls_target = encode_target(boxes[i], labels[i], input_size=(w,h))
File "H:\PyCharm project\retinanet\utils\util.py", line 39, in encode_target
anchor_bboxes=enumerate_all_anchors(input_size)
File "H:\PyCharm project\retinanet\utils\util.py", line 64, in enumerate_all_anchors
feature_map_size=[(input_size/pow(2.,i+3)).ceil() for i in range(num_feature)] # p3-p7feature_map size
File "H:\PyCharm project\retinanet\utils\util.py", line 64, in
feature_map_size=[(input_size/pow(2.,i+3)).ceil() for i in range(num_feature)] # p3-p7feature_map size
RuntimeError: ceil_vml_cpu not implemented for 'Long'
Process finished with exit code 1
Anyone can help me?Thanks
I have this error when run train.py: Traceback (most recent call last): File "H:/PyCharm project/retinanet/train.py", line 101, in
train(epoch)
File "H:/PyCharm project/retinanet/train.py", line 55, in train
for batch_idx, (inputs, loc_targets, cls_targets) in enumerate(trainloader):
File "D:\Python\lib\site-packages\torch\utils\data\dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "D:\Python\lib\site-packages\torch\utils\data\dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
File "D:\Python\lib\site-packages\torch\utils\data_utils\worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "H:\PyCharm project\retinanet\utils\util.py", line 28, in collate_fn
loc_target, cls_target = encode_target(boxes[i], labels[i], input_size=(w,h))
File "H:\PyCharm project\retinanet\utils\util.py", line 39, in encode_target
anchor_bboxes=enumerate_all_anchors(input_size)
File "H:\PyCharm project\retinanet\utils\util.py", line 64, in enumerate_all_anchors
feature_map_size=[(input_size/pow(2.,i+3)).ceil() for i in range(num_feature)] # p3-p7feature_map size
File "H:\PyCharm project\retinanet\utils\util.py", line 64, in
feature_map_size=[(input_size/pow(2.,i+3)).ceil() for i in range(num_feature)] # p3-p7feature_map size
RuntimeError: ceil_vml_cpu not implemented for 'Long'
Process finished with exit code 1 Anyone can help me?Thanks