Thank you for your good research. And thank you for revealing the code quickly.
I tried to implement your code in the same environment.
However, using a single GPU environment, the batch have become 1/2 size. And I got the following results.
edit config line 4 : GPUS: (1,) (in my environment using 0:3080, 1:2080TI)
I also encountered an error in the posetrack 2017 test dataset.
2021-04-02 14:23:13 [engine.core.function] INFO: test: [3100/5462] Time 1.659 (1.713) Data 0.027s (0.083s) Accuracy 0.000 (0.006)
2021-04-02 14:25:59 [engine.core.function] INFO: test: [3200/5462] Time 1.659 (1.712) Data 0.027s (0.081s) Accuracy 0.000 (0.006)
Traceback (most recent call last):
File "run.py", line 33, in <module>
main()
File "run.py", line 29, in main
runner.launch()
File "/DCPose/engine/defaults/runner.py", line 63, in launch
evaluator.exec()
File "/DCPose/engine/defaults/evaluator.py", line 20, in exec
self.eval()
File "/DCPose/engine/defaults/evaluator.py", line 73, in eval
phase=self.phase)
File "/DCPose/engine/core/function.py", line 165, in eval
input_x, input_sup_A, input_sup_B, target_heatmaps, target_heatmaps_weight, meta = next(self.dataloader_iter)
File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 345, in __next__
data = self._next_data()
File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 856, in _next_data
return self._process_data(data)
File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 881, in _process_data
data.reraise()
File "/opt/conda/lib/python3.7/site-packages/torch/_utils.py", line 394, in reraise
raise self.exc_type(msg)
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/opt/conda/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/DCPose/datasets/zoo/posetrack/PoseTrack.py", line 100, in __getitem__
return self._get_spatiotemporal_window(data_item)
File "/DCPose/datasets/zoo/posetrack/PoseTrack.py", line 166, in _get_spatiotemporal_window
self.logger.error(error_msg)
AttributeError: 'PoseTrack' object has no attribute 'logger'