Spaces:
Running
on
T4
Running
on
T4
| # -------------------------------------------------------- | |
| # SiamMask | |
| # Licensed under The MIT License | |
| # Written by Qiang Wang (wangqiang2015 at ia.ac.cn) | |
| # -------------------------------------------------------- | |
| import torch.nn as nn | |
| class Mask(nn.Module): | |
| def __init__(self): | |
| super(Mask, self).__init__() | |
| def forward(self, z_f, x_f): | |
| raise NotImplementedError | |
| def template(self, template): | |
| raise NotImplementedError | |
| def track(self, search): | |
| raise NotImplementedError | |
| def param_groups(self, start_lr, feature_mult=1): | |
| params = filter(lambda x:x.requires_grad, self.parameters()) | |
| params = [{'params': params, 'lr': start_lr * feature_mult}] | |
| return params | |