Model
UNet_Industrial TensorFlow checkpoint trained with AMP
Use the NGC CLI to download:
Copied!
2 Versions
19.08.0Selected08/21/2020 3:18 AM UTC327.09 MBAccuracy: 00 EpochsBatch Size: 0GPU: V100 Copied!
19.08.0Selected
08/21/2020 3:18 AM UTC327.09 MBAccuracy: 00 EpochsBatch Size: 0GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| config | default |
performance
| Key | Value |
|---|---|
| accuracy | >96% |
training
| Key | Value |
|---|---|
| bs | 16 |
| dataset | DAGM2007 |
| steps | 2500 |
| training_precision | AMP |
08/21/2020 3:17 AM UTC132.53 MBAccuracy: 072 EpochsBatch Size: 2GPU: V100 Copied!
08/21/2020 3:17 AM UTC132.53 MBAccuracy: 072 EpochsBatch Size: 2GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| type | base |
performance
| Key | Value |
|---|---|
| eval.IoU_THS_0.05 | 0.971 |
| eval.IoU_THS_0.125 | 0.976 |
| eval.IoU_THS_0.25 | 0.976 |
| eval.IoU_THS_0.5 | 0.976 |
| eval.IoU_THS_0.75 | 0.976 |
| eval.IoU_THS_0.85 | 0.976 |
| eval.IoU_THS_0.95 | 0.976 |
| eval.IoU_THS_0.99 | 0.976 |
training
| Key | Value |
|---|---|
| batch_size | 2 |
| dataset | DAGM 2007 |
| epochs | 72 |
| learning_rate | 0.0001 |
| steps | 2500 |
| training_precision | AMP |