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

wip finishing preparation

parent ad9e70f0
......@@ -40,7 +40,7 @@ class SRDataset(data.Dataset):
def _random_crop(self, lr, hr):
h = hr.shape[1] - self.crop[0]
w = hr.shape[2] - self.crop[1]
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 @@
"from core.base import BaseModule\n",
"from core.modules import UnetBT, BaseUpsampler\n",
"from core.module_unet import UnetCascade\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