update readme / configs

valeoai · Dec 12, 2022 · ee2edec · ee2edec
1 parent 24eaaee
commit ee2edec
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -199,8 +199,6 @@ Detection | KITTI3D | SECOND/PVRCNN | download | trained for KITTI3D |
 
 Here are links to pre-trained models
 
-## Performances
-
 ## Acknowledgments
 
 This project would not have been possible without many community ressources and repositories. Among them:

diff --git a/also_selfsup/configs/cfg/kitti360_second.yaml b/also_selfsup/configs/cfg/kitti360_second.yaml
@@ -15,7 +15,7 @@ inputs: ["pos", "intensities"]
 
 # optimization
 training:
-  max_epochs: 100
+  max_epochs: 75
   batch_size: 8
   val_interval: 5
 

diff --git a/also_selfsup/configs/cfg/kitti3d_second.yaml b/also_selfsup/configs/cfg/kitti3d_second.yaml
@@ -15,7 +15,7 @@ inputs: ["pos", "intensities"]
 
 # optimization
 training:
-  max_epochs: 100
+  max_epochs: 500
   batch_size: 8
   val_interval: 5
 

diff --git a/also_selfsup/configs/cfg/once_raw_small_second.yaml b/also_selfsup/configs/cfg/once_raw_small_second.yaml
@@ -15,7 +15,7 @@ inputs: ["pos", "intensities"]
 
 # optimization
 training:
-  max_epochs: 50
+  max_epochs: 40
   batch_size: 8
   val_interval: 5