updates the tb logs dir
Browse files
train.py
CHANGED
|
@@ -26,7 +26,7 @@ model = DRModel(
|
|
| 26 |
)
|
| 27 |
|
| 28 |
# Init logger
|
| 29 |
-
logger = TensorBoardLogger(save_dir="
|
| 30 |
# Init callbacks
|
| 31 |
checkpoint_callback = ModelCheckpoint(
|
| 32 |
monitor="val_loss",
|
|
|
|
| 26 |
)
|
| 27 |
|
| 28 |
# Init logger
|
| 29 |
+
logger = TensorBoardLogger(save_dir="logs",name="")
|
| 30 |
# Init callbacks
|
| 31 |
checkpoint_callback = ModelCheckpoint(
|
| 32 |
monitor="val_loss",
|