a2d2/a2d2_dataset.py (3 lines): - line 126: # TODO: consider use torch.Tensor only - line 622: # TODO: refactor this function - line 627: # TODO: remove the hack of yaw container_training/train.py (1 line): - line 212: # TODO: ugly workaround to judge whether we are training det or seg model docker/train.py (1 line): - line 212: # TODO: ugly workaround to judge whether we are training det or seg model