Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Distributed training [SemSeg] #530

Open
wants to merge 55 commits into
base: main
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
55 commits
Select commit Hold shift + click to select a range
2f0e72b
add dataparallel
sanskar107 Aug 12, 2021
306de5b
add dataparallel class
sanskar107 Aug 12, 2021
b953254
fix bugs
sanskar107 Aug 13, 2021
4dccfc0
objdet multigpu
sanskar107 Aug 24, 2021
2779339
rename scatter
sanskar107 Aug 27, 2021
48a101f
fix objdet
sanskar107 Sep 24, 2021
53479c2
remove comments
sanskar107 Sep 24, 2021
388f948
Merge branch 'dev' into sanskar/multi_gpu
sanskar107 Jan 10, 2022
6139e66
fix cam_img matrix
sanskar107 Jan 10, 2022
d9e1564
add distributed training
sanskar107 Jan 17, 2022
ef0e440
parllel validation
sanskar107 Jan 18, 2022
29b9729
update config
sanskar107 Feb 7, 2022
672248d
gather in run_valid
sanskar107 Feb 8, 2022
620b35c
fix preprocessing
sanskar107 Feb 9, 2022
1c64d65
add shuffle
sanskar107 Feb 18, 2022
a22a8dc
fix rng
sanskar107 Feb 18, 2022
685dd3a
remove customparallel
sanskar107 Feb 18, 2022
bfaa4a2
reset semseg distributed training
sanskar107 Feb 18, 2022
10164f9
Merge branch 'dev' into sanskar/multi_gpu
sanskar107 Feb 18, 2022
6d58cd1
change config
sanskar107 Feb 22, 2022
51a16c3
fix lgtm
sanskar107 Feb 22, 2022
50133bb
address reviews (1)
sanskar107 Mar 25, 2022
1ed11c8
fix model.module....
sanskar107 Mar 25, 2022
39b3117
add semseg labels
sanskar107 Mar 28, 2022
142df86
add waymo semseg preprocessing
sanskar107 Mar 28, 2022
5391b1c
add waymo semseg dataset
sanskar107 Mar 31, 2022
eb3b551
remove dataparallel
sanskar107 Apr 5, 2022
bb7a715
change sampler
sanskar107 Apr 12, 2022
6f6913e
Merge branch 'sanskar/waymo_semseg' into sanskar/dist_semseg
sanskar107 Apr 12, 2022
d35d2f1
add val sampler
sanskar107 Apr 12, 2022
db1128c
add config files
sanskar107 Apr 12, 2022
d718080
fix waymo semseg
sanskar107 Apr 18, 2022
8e8d4f8
fix optim
sanskar107 Apr 18, 2022
3997833
improve tqdm
sanskar107 Apr 18, 2022
5a4d385
gather metric among process
sanskar107 Apr 18, 2022
df76487
enable multi node training
sanskar107 Apr 27, 2022
6fc34b1
add nuscenes semantic
sanskar107 May 4, 2022
80d0040
fix lidarseg
sanskar107 May 4, 2022
670ef48
add megaloader
sanskar107 May 10, 2022
e211235
Merge branch 'dev' into sanskar/dist_semseg
sanskar107 May 11, 2022
1fdf0d1
add megamodel
sanskar107 May 11, 2022
b5c5296
update linear layers
sanskar107 May 13, 2022
e4c4441
add new pipeline
sanskar107 May 25, 2022
ea5f303
add nuscenes semseg
sanskar107 May 25, 2022
4ff5fbe
add kitti360
sanskar107 Jun 3, 2022
43c39c7
fixes
sanskar107 Jun 7, 2022
1803d77
apply style
sanskar107 Jun 7, 2022
c8f4589
fix logging
sanskar107 Jun 9, 2022
5a4371a
add distributed semseg
sanskar107 Sep 20, 2022
8c60c40
Merge branch 'dev' into sanskar/dist_semseg
sanskar107 Sep 20, 2022
40f888f
apply-style
sanskar107 Sep 20, 2022
375bb8f
remove megamodel
sanskar107 Sep 20, 2022
50c8c20
fix docstring
sanskar107 Sep 20, 2022
5c2ba32
address lgtm
sanskar107 Sep 22, 2022
e971747
update preprocess
sanskar107 Sep 22, 2022
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update config
sanskar107 committed Feb 7, 2022
commit 29b972906a099996541e0c897737cbc688b65010
14 changes: 7 additions & 7 deletions ml3d/configs/pointpillars_waymo.yml
Original file line number Diff line number Diff line change
@@ -37,7 +37,7 @@ model:

scatter:
in_channels: 64
output_shape: [468, 468]
output_shape: [520, 520]

backbone:
in_channels: 64
@@ -57,14 +57,14 @@ model:
nms_pre: 4096
score_thr: 0.1
ranges: [
[-74.88, -74.88, -0.0345, 74.88, 74.88, -0.0345],
[-74.88, -74.88, -0.1188, 74.88, 74.88, -0.1188],
[-74.88, -74.88, 0, 74.88, 74.88, 0],
[-80, -80, 1.142, 85, 85, 1.142],
[-80, -80, 1.139, 85, 85, 1.139],
[-80, -80, 1.149, 85, 85, 1.149],
]
sizes: [
[2.08, 4.73, 1.77], # car
[0.84, 1.81, 1.77], # cyclist
[0.84, 0.91, 1.74] # pedestrian
[1.98, 4.50, 1.96], # VEHICLE
[0.91, 1.94, 1.78], # CYCLIST
[0.84, 0.91, 1.70] # PEDESTRIAN
]
dir_offset: 0.7854
rotations: [0, 1.57]