feat: add deterministic
Browse files
train.py
CHANGED
|
@@ -56,6 +56,7 @@ trainer = ptl.Trainer(
|
|
| 56 |
enable_checkpointing=True,
|
| 57 |
log_every_n_steps=log_every_n_steps,
|
| 58 |
strategy=strategy,
|
|
|
|
| 59 |
)
|
| 60 |
|
| 61 |
model = ResNet18Regressor()
|
|
|
|
| 56 |
enable_checkpointing=True,
|
| 57 |
log_every_n_steps=log_every_n_steps,
|
| 58 |
strategy=strategy,
|
| 59 |
+
deterministic=True,
|
| 60 |
)
|
| 61 |
|
| 62 |
model = ResNet18Regressor()
|