ljaljushkin

import torch
from torch import nn
from torch.optim import SGD
from torch.utils import data
from torch.utils.data import DataLoader


class UNet(nn.Module):
    def __init__(
            self,