marlenezw commited on
Commit
fc4c286
·
1 Parent(s): b6c3fb5

Update MakeItTalk/thirdparty/AdaptiveWingLoss/core/models.py

Browse files
MakeItTalk/thirdparty/AdaptiveWingLoss/core/models.py CHANGED
@@ -3,6 +3,7 @@ import torch.nn as nn
3
  import torch.nn.functional as F
4
  import math
5
  from core.coord_conv import CoordConvTh
 
6
 
7
 
8
  def conv3x3(in_planes, out_planes, strd=1, padding=1,
 
3
  import torch.nn.functional as F
4
  import math
5
  from core.coord_conv import CoordConvTh
6
+ from thirdparty.AdaptiveWingLoss.core.coord_conv import CoordConvTh
7
 
8
 
9
  def conv3x3(in_planes, out_planes, strd=1, padding=1,