text: # Copyright (C) 2021 NVIDIA CORPORATION & AFFILIATES.  All rights reserved.
  #
  # This work is made available under the Nvidia Source Code License-NC.
  # To view a copy of this license, check out LICENSE.md
  # The file is duplicated from https://github.com/NVIDIA/flownet2-pytorch
  # with some modifications.
  