Searched refs:include_loss_args (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/benchmarks/dynamo/ |
| D | huggingface.py | 172 model_cls, model, model_name, bs, device, include_loss_args=False argument
231 if include_loss_args:
435 model_cls, model, model_name, batch_size, device, include_loss_args=True
|