Commit 19187d3d authored by Matej Choma's avatar Matej Choma

wip finishing preparation

parent ad9e70f0
import torch
from torch import nn
from core.base import base_conv2d, base_upsample, ResidualCatConv
from core.base import BaseModule, base_conv2d, base_upsample, ResidualCatConv
import math
class ChannelAttention(nn.Module):
......
This source diff could not be displayed because it is too large. You can view the blob instead.
......@@ -23,6 +23,7 @@
"\n",
"from core.base import BaseModule\n",
"from core.modules import UnetBT, BaseUpsampler\n",
"from core.module_unet import UnetCascade\n",
"\n",
"import numpy as np"
]
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment