Outputs = parallel_apply(replicas, inputs, module_kwargs, used_device_ids)įile "/opt/miniconda/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 86, in parallel_applyįile "/opt/miniconda/lib/python3.7/site-packages/torch/_utils.py", line 461, in reraise mean() # / self.accum_gradįile "/opt/miniconda/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 231, in data_parallel Train_stats = train_epoch(train_loader, model, optimizer, epoch)ĭata_parallel(model, (fbank, seq_lens, tokens, language), range(args.ngpu))#. Train(dataloaders, model, optimizer, save_path)
0 kommentar(er)
