A mistake in val_apollo.py

[Sillysillysilly]

When I try to run the code:"python3 val_apollo.py"
I get a run error:
NameError: name 'Dataset' is not defined

After comparing with val_openlane.py
I think there should be a missing line of code in val_apollo.py which is:
"from torch.utils.data import Dataset,DataLoader"

[qinjian623]

Yes, you're right.

Sorry for this mistake.