diff --git a/model.py b/model.py index 2824779..3083762 100644 --- a/model.py +++ b/model.py @@ -1,7 +1,7 @@ # This file is not used by the tracking application and currently outdated import torch import torch.nn as nn -import geffnet.mobilenetv3 # geffnet.mobilenetv3._gen_mobilenet_v3 needs to be patched to return the parameters instead of instantiating the network +import geffnet.mobilenetv3 # geffnet.mobilenetv3._gen_mobilenet_v3 needs to be patched to return the parameters instead of instantiating the network, based on commit c450c12ae6ffb1757f62dde3c2765da3c10f6def from geffnet.efficientnet_builder import round_channels class DSConv2d(nn.Module):