Unverified Commit 77cc4ef5 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

[easy] Expand usage logging to video model (#4737)



Summary: As titled.

Reviewed By: sallysyw

Differential Revision: D31813675

fbshipit-source-id: 8ff879be172530bcc41aae5d71c67e1e90f09cbf
Co-authored-by: default avatarKai Zhang <kaizh@fb.com>
parent f28d93b0
......@@ -4,7 +4,7 @@ import torch.nn as nn
from torch import Tensor
from ..._internally_replaced_utils import load_state_dict_from_url
from ...utils import _log_api_usage_once
__all__ = ["r3d_18", "mc3_18", "r2plus1d_18"]
......@@ -209,6 +209,7 @@ class VideoResNet(nn.Module):
zero_init_residual (bool, optional): Zero init bottleneck residual BN. Defaults to False.
"""
super(VideoResNet, self).__init__()
_log_api_usage_once(self)
self.inplanes = 64
self.stem = stem()
......
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