Unverified Commit c47c9196 authored by Jerry Jiarui XU's avatar Jerry Jiarui XU Committed by GitHub
Browse files

[Docs] Add header for files (#1181)

* [Docs] Add header for files

* change to OpenMMLab

* add headers for .cpp, .cu, .h, .cuh

* replace Open-MMLab with OpenMMLab
parent 04daea42
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np import numpy as np
import torch import torch
......
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
from torch.autograd import Function from torch.autograd import Function
......
# Copyright (c) OpenMMLab. All rights reserved.
import torch.nn as nn import torch.nn as nn
from torch.autograd import Function from torch.autograd import Function
......
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
from torch.autograd import Function from torch.autograd import Function
......
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
......
# Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.distributed as dist import torch.distributed as dist
import torch.nn.functional as F import torch.nn.functional as F
......
# Copyright (c) OpenMMLab. All rights reserved.
# Code reference from "Temporal Interlacing Network" # Code reference from "Temporal Interlacing Network"
# https://github.com/deepcs233/TIN/blob/master/cuda_shift/rtc_wrap.py # https://github.com/deepcs233/TIN/blob/master/cuda_shift/rtc_wrap.py
# Hao Shao, Shengju Qian, Yu Liu # Hao Shao, Shengju Qian, Yu Liu
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .collate import collate from .collate import collate
from .data_container import DataContainer from .data_container import DataContainer
from .data_parallel import MMDataParallel from .data_parallel import MMDataParallel
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
from torch.nn.parallel._functions import _get_stream from torch.nn.parallel._functions import _get_stream
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from collections.abc import Mapping, Sequence from collections.abc import Mapping, Sequence
import torch import torch
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import functools import functools
import torch import torch
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from itertools import chain from itertools import chain
from torch.nn.parallel import DataParallel from torch.nn.parallel import DataParallel
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
from torch.nn.parallel.distributed import (DistributedDataParallel, from torch.nn.parallel.distributed import (DistributedDataParallel,
_find_tensors) _find_tensors)
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
import torch.distributed as dist import torch.distributed as dist
import torch.nn as nn import torch.nn as nn
......
# Copyright (c) OpenMMLab. All rights reserved.
from torch.nn.parallel import DataParallel, DistributedDataParallel from torch.nn.parallel import DataParallel, DistributedDataParallel
from mmcv.utils import Registry from mmcv.utils import Registry
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import torch import torch
from torch.nn.parallel._functions import Scatter as OrigScatter from torch.nn.parallel._functions import Scatter as OrigScatter
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .registry import MODULE_WRAPPERS from .registry import MODULE_WRAPPERS
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .base_module import BaseModule, ModuleList, Sequential from .base_module import BaseModule, ModuleList, Sequential
from .base_runner import BaseRunner from .base_runner import BaseRunner
from .builder import RUNNERS, build_runner from .builder import RUNNERS, build_runner
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import copy import copy
import warnings import warnings
from abc import ABCMeta from abc import ABCMeta
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import copy import copy
import logging import logging
import os.path as osp import os.path as osp
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment