snicolau commited on
Commit
bddf50c
Β·
verified Β·
1 Parent(s): 0570614

Update lama/saicinpainting/training/modules/multidilated_conv.py

Browse files
lama/saicinpainting/training/modules/multidilated_conv.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  import torch.nn as nn
3
  import random
4
- from saicinpainting.training.modules.depthwise_sep_conv import DepthWiseSeperableConv
5
 
6
  class MultidilatedConv(nn.Module):
7
  def __init__(self, in_dim, out_dim, kernel_size, dilation_num=3, comb_mode='sum', equal_dim=True,
 
1
  import torch
2
  import torch.nn as nn
3
  import random
4
+ from lama.saicinpainting.training.modules.depthwise_sep_conv import DepthWiseSeperableConv
5
 
6
  class MultidilatedConv(nn.Module):
7
  def __init__(self, in_dim, out_dim, kernel_size, dilation_num=3, comb_mode='sum', equal_dim=True,