Open TrefoIV opened 1 year ago
When using autocast as in line engine.py#L51 with device type "cpu" a warning is raised.
Using torch.autocast(device_type, enabled ...) resolve the warning.
torch.autocast(device_type, enabled ...)
When using autocast as in line engine.py#L51 with device type "cpu" a warning is raised.
Using
torch.autocast(device_type, enabled ...)
resolve the warning.