Skip to content

Commit 6d40f83

Browse files
went back to temp model
1 parent eb5a2b2 commit 6d40f83

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

Simulateur/config.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@
1111
lidar_max_range = 12.0
1212
device = "cuda" if is_available() else "cpu"
1313

14-
context_size = 1
15-
lidar_horizontal_resolution = 1024 # DON'T CHANGE THIS VALUE PLS
16-
camera_horizontal_resolution = 1024 # DON'T CHANGE THIS VALUE PLS
14+
context_size = 128
15+
lidar_horizontal_resolution = 128 # DON'T CHANGE THIS VALUE PLS
16+
camera_horizontal_resolution = 128 # DON'T CHANGE THIS VALUE PLS
1717

1818
B_DEBUG = False

Simulateur/launch_train_multiprocessing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def make_env(rank: int):
125125

126126
envs = SubprocVecEnv([lambda rank=rank : make_env(rank) for rank in range(n_simulations)])
127127

128-
ExtractorClass = CNN1DResNetExtractor
128+
ExtractorClass = TemporalResNetExtractor
129129

130130
policy_kwargs = dict(
131131
features_extractor_class=ExtractorClass,

0 commit comments

Comments
 (0)