riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 19:21:31 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

removed redundant arg in prepare_inputs (#6614)

Browse Source 
* removed redundant arg in prepare_inputs

* made same change in prediction_loop
This commit is contained in:
Prajjwal Bhargava 2020-08-20 17:53:35 +05:30 committed by GitHub
parent cabfdfafc0
commit 33bf426498
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/trainer.py
View File

@ -705,7 +705,7 @@ class Trainer:
print(output)
def _prepare_inputs(
self, inputs: Dict[str, Union[torch.Tensor, Any]], model: nn.Module
self, inputs: Dict[str, Union[torch.Tensor, Any]]
) -> Dict[str, Union[torch.Tensor, Any]]:
"""
Prepare :obj:`inputs` before feeding them to the model, converting them to tensors if they are not already and
@ -746,7 +746,7 @@ class Trainer:
return self._training_step(model, inputs, self.optimizer)
model.train()
inputs = self._prepare_inputs(inputs, model)
inputs = self._prepare_inputs(inputs)
if self.args.fp16 and _use_native_amp:
with autocast():
@ -1071,7 +1071,7 @@ class Trainer:
"""
has_labels = any(inputs.get(k) is not None for k in ["labels", "lm_labels", "masked_lm_labels"])
inputs = self._prepare_inputs(inputs, model)
inputs = self._prepare_inputs(inputs)
with torch.no_grad():
outputs = model(**inputs)