Saving weights and logs of step 1000
Browse files
events.out.tfevents.1642699155.t1v-n-ccbf3e94-w-0.1759255.3.v2
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:649d3492d28b9cad2af7bf4164a44419b286e13c1d3f10f62101eb4fc5dd5ebf
|
3 |
+
size 147136
|
flax_model.msgpack
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 498796983
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:63984c1d7b5b86c5f5bc02d8c28737e0737c41604565915da6709493423696a2
|
3 |
size 498796983
|
run_mlm_flax.py
CHANGED
@@ -557,8 +557,8 @@ def main():
|
|
557 |
# Enable Weight&Biases
|
558 |
import wandb
|
559 |
wandb.init(
|
560 |
-
entity='
|
561 |
-
project='
|
562 |
sync_tensorboard=False,
|
563 |
)
|
564 |
wandb.config.update(training_args)
|
@@ -566,12 +566,15 @@ def main():
|
|
566 |
wandb.config.update(data_args)
|
567 |
|
568 |
from flax.metrics.tensorboard import SummaryWriter
|
569 |
-
|
570 |
summary_writer = SummaryWriter(log_dir=Path(training_args.output_dir))
|
|
|
571 |
except ImportError as ie:
|
572 |
-
|
|
|
|
|
|
|
573 |
logger.warning(
|
574 |
-
f"Unable to display metrics through
|
575 |
)
|
576 |
else:
|
577 |
logger.warning(
|
|
|
557 |
# Enable Weight&Biases
|
558 |
import wandb
|
559 |
wandb.init(
|
560 |
+
entity='undefined',
|
561 |
+
project='undefined',
|
562 |
sync_tensorboard=False,
|
563 |
)
|
564 |
wandb.config.update(training_args)
|
|
|
566 |
wandb.config.update(data_args)
|
567 |
|
568 |
from flax.metrics.tensorboard import SummaryWriter
|
|
|
569 |
summary_writer = SummaryWriter(log_dir=Path(training_args.output_dir))
|
570 |
+
|
571 |
except ImportError as ie:
|
572 |
+
from flax.metrics.tensorboard import SummaryWriter
|
573 |
+
summary_writer = SummaryWriter(log_dir=Path(training_args.output_dir))
|
574 |
+
|
575 |
+
has_tensorboard = True
|
576 |
logger.warning(
|
577 |
+
f"Unable to display metrics through Wandb because some package are not installed: {ie}"
|
578 |
)
|
579 |
else:
|
580 |
logger.warning(
|