ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: removed now useless Sequential

Browse source 
Former-commit-id: 4b97d6d3bef3814f7cfc6c6ceca7833105b4211a [formerly 29f131cf14a73b22427f4ef2318def6870cbe7e0]
Former-commit-id: db11e7b2ef2ba39fe11e69251731ea2489d3be31
This commit is contained in:
Laurent Fainsin 2022-07-01 10:33:35 +02:00
parent 2ab95734e4
commit 31ceb97996
2 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/train.py
View file

@ -22,7 +22,7 @@ def main():
wandb.init(
project="U-Net",
config=dict(
DIR_TRAIN_IMG="/home/lilian/data_disk/lfainsin/smolval2017",
DIR_TRAIN_IMG="/home/lilian/data_disk/lfainsin/val2017",
DIR_VALID_IMG="/home/lilian/data_disk/lfainsin/smoltrain2017/",
DIR_SPHERE_IMG="/home/lilian/data_disk/lfainsin/spheres/Images/",
DIR_SPHERE_MASK="/home/lilian/data_disk/lfainsin/spheres/Masks/",
@ -31,7 +31,7 @@ def main():
N_CLASSES=1,
AMP=True,
PIN_MEMORY=True,
BENCHMARK=False,
BENCHMARK=True,
DEVICE="cuda",
WORKERS=8,
EPOCHS=5,

5
src/unet/blocks.py
View file

@ -70,10 +70,7 @@ class OutConv(nn.Module):
def __init__(self, in_channels, out_channels):
super(OutConv, self).__init__()
self.conv = nn.Sequential(
nn.Conv2d(in_channels, out_channels, kernel_size=1),
# nn.Sigmoid(),
)
self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=1)
def forward(self, x):
return self.conv(x)