Model
SIM TensorFlow2 checkpoint trained on Amazon Books 2014 Dataset prebatched with size of 4096
Use the NGC CLI to download:
Copied!
2 Versions
22.01.0_fp32Selected12/05/2022 4:39 PM UTC243.8 MBAccuracy: 03 EpochsBatch Size: 131072GPU: V100 Copied!
22.01.0_fp32Selected
12/05/2022 4:39 PM UTC243.8 MBAccuracy: 03 EpochsBatch Size: 131072GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| type | base |
performance
| Key | Value |
|---|---|
| AUC | 0.8117 |
training
| Key | Value |
|---|---|
| batch_size | 131072 |
| training_precision | FP32 |
| dataset | Amazon Books 2014 Dataset |
| learning_rate | 0.01 |
| epochs | 3 |
12/05/2022 4:38 PM UTC243.8 MBAccuracy: 03 EpochsBatch Size: 131072GPU: V100 Copied!
12/05/2022 4:38 PM UTC243.8 MBAccuracy: 03 EpochsBatch Size: 131072GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| type | base |
performance
| Key | Value |
|---|---|
| AUC | 0.8221 |
training
| Key | Value |
|---|---|
| batch_size | 131072 |
| training_precision | AMP |
| dataset | Amazon Books 2014 Dataset |
| learning_rate | 0.01 |
| epochs | 3 |