diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp index 4ee33e9ab..7c3da5c43 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py index ca5afec31..fdd84245b 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py @@ -1192,6 +1192,9 @@ def main(): logging.info("save arguments to config.yaml...") save_args(args) + if args.wandb: wb = wandb.init(project="d2v-T", entity="dohe0342", config=vars(args)) + else: wb = None + world_size = args.world_size assert world_size >= 1 if world_size > 1: