Spaces:
Running
Running
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,
|