You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The script should run without errors with autocast enabled, as otherwise training with AMP is not available.
Output of the script above:
Running with autocast setting: {'dtype': torch.float16, 'enabled': True}
Index put requires the source and destination dtypes match, got Float for the destination and Half for the source.
Running with autocast setting: {'dtype': torch.bfloat16, 'enabled': True}
Index put requires the source and destination dtypes match, got Float for the destination and BFloat16 for the source.
Running with autocast setting: {'enabled': False}
torch.Size([16, 300, 288])
System Info
transformers
version: 4.40.2Who can help?
@ArthurZucker
Information
Tasks
examples
folder (such as GLUE/SQuAD, ...)Reproduction
Expected behavior
The script should run without errors with autocast enabled, as otherwise training with AMP is not available.
Output of the script above:
Expected output:
The text was updated successfully, but these errors were encountered: