Update README.md
Browse files
README.md
CHANGED
|
@@ -71,7 +71,7 @@ data together. It also offers the option to flatten the data and provide a `bat
|
|
| 71 |
### Pretraining
|
| 72 |
|
| 73 |
The model was trained on a node of 4xA40 (48 GB) for 10 epochs (~2 weeks). See the
|
| 74 |
-
[training code](https://github.com/
|
| 75 |
details.
|
| 76 |
|
| 77 |
## Evaluation results
|
|
|
|
| 71 |
### Pretraining
|
| 72 |
|
| 73 |
The model was trained on a node of 4xA40 (48 GB) for 10 epochs (~2 weeks). See the
|
| 74 |
+
[training code](https://github.com/VectorInstitute/AtomGen) for all hyperparameters
|
| 75 |
details.
|
| 76 |
|
| 77 |
## Evaluation results
|