Skip to content
Snippets Groups Projects
Commit 6e505699 authored by s.islam's avatar s.islam
Browse files

Real Data, 9 channels + Retardation, resnet18, mask, WLoss, 522

parent d37f462a
Branches master
No related tags found
No related merge requests found
......@@ -8,14 +8,13 @@ import numpy as np
import segmentation_models_pytorch as smp
from torch import Tensor, nn
from pl_bolts.models.gans.dcgan.components import DCGANDiscriminator
#from pl_bolts.models.gans.dcgan.components import DCGANDiscriminator
'''
class DCGANDiscriminator(nn.Module):
def __init__(self, feature_maps: int,
image_channels: int) -> None: # This part has been modified for Wasserstein loss
......@@ -61,7 +60,7 @@ class DCGANDiscriminator(nn.Module):
def forward(self, x: Tensor) -> Tensor:
return self.disc(x).view(-1, 1).squeeze(1)
'''
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment