transformers
3280b132 - Fixing a bug when MlFlow try to log a torch.tensor (#29932)

Comment changes are shownComment changes are hidden
Commit
1 year ago
Fixing a bug when MlFlow try to log a torch.tensor (#29932) * Update integration_utils.py Add the case where a tensor with one element is log with Mlflow * Update src/transformers/integrations/integration_utils.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Update integration_utils.py add a whitespace --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Parents
  • src/transformers/integrations
    • File
      integration_utils.py