Colab version:

Further readings:

• LR warm-up and useful tricks - article

Reference:based on TorchVision 0.3 object detection tutorial