sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
223 stars 75 forks source link

Autocast warning #37

Open TrefoIV opened 1 year ago

TrefoIV commented 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.