SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

remove check for main process for trackers initialization (#18706)

Browse Source
This commit is contained in:
Atharva Ingle
2022-08-22 20:46:27 +05:30
committed by GitHub
parent 0f257a8774
commit d90a36d192
9 changed files with 45 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/pytorch/semantic-segmentation/run_semantic_segmentation_no_trainer.py
View File

@@ -512,14 +512,12 @@ def main():
metric = evaluate.load("mean_iou")
# We need to initialize the trackers we use, and also store our configuration.
# We initialize the trackers only on main process because `accelerator.log`
# only logs on main process and we don't want empty logs/runs on other processes.
# The trackers initializes automatically on the main process.
if args.with_tracking:
if accelerator.is_main_process:
experiment_config = vars(args)
# TensorBoard cannot log Enums, need the raw value
experiment_config["lr_scheduler_type"] = experiment_config["lr_scheduler_type"].value
accelerator.init_trackers("semantic_segmentation_no_trainer", experiment_config)
experiment_config = vars(args)
# TensorBoard cannot log Enums, need the raw value
experiment_config["lr_scheduler_type"] = experiment_config["lr_scheduler_type"].value
accelerator.init_trackers("semantic_segmentation_no_trainer", experiment_config)
# Train!
total_batch_size = args.per_device_train_batch_size * accelerator.num_processes * args.gradient_accumulation_steps