Update README.md
Browse files
README.md
CHANGED
|
@@ -141,6 +141,15 @@ For inference, a preprocessing as close as possible should be used. Refer to the
|
|
| 141 |
|
| 142 |
#### Speeds, Sizes, Times
|
| 143 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 144 |
|
| 145 |
## Evaluation
|
| 146 |
|
|
|
|
| 141 |
|
| 142 |
#### Speeds, Sizes, Times
|
| 143 |
|
| 144 |
+
The **FRACTAL-LidarHD_7cl_randlanet** model was trained on an in-house HPC cluster.
|
| 145 |
+
6 V100 GPUs were used (2 nodes, 3 GPUS per node). With this configuration the approximate learning time is 30 minutes per epoch.
|
| 146 |
+
|
| 147 |
+
The model was obtained for num_epoch=21 with corresponding val_loss=0.112.
|
| 148 |
+
|
| 149 |
+
<div style="position: relative; text-align: center;">
|
| 150 |
+
<p style="margin: 0;">TRAIN loss</p>
|
| 151 |
+
<img src="FRACTAL-LidarHD_7cl_randlanet-train_val_losses.excalidraw.png" alt="train and val losses" style="width: 60%; display: block; margin: 0 auto;"/>
|
| 152 |
+
</div>
|
| 153 |
|
| 154 |
## Evaluation
|
| 155 |
|