update the support for the Apollo dataset

OpenDriveLab · Apr 8, 2023 · 7f7715a · 7f7715a
1 parent 41e06e8
commit 7f7715a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/main_persformer.py b/main_persformer.py
@@ -15,7 +15,7 @@
 # ==============================================================================
 
 import os
-from config import persformer_openlane, persformer_once
+from config import persformer_openlane, persformer_once, persformer_apollo
 from utils.utils import *
 from experiments.ddp import *
 from experiments.runner import *
@@ -24,7 +24,9 @@
 def main():
     parser = define_args() # args in utils.py
     args = parser.parse_args()
+
     # specify dataset and model config
+    # persformer_apollo.config(args)
     # persformer_once.config(args)
     persformer_openlane.config(args)
     # initialize distributed data parallel set