Skip to content

Commit cac2d01

Browse files
StephanieLarocquenotoraptor
authored andcommitted
adde training loop link
1 parent ce89838 commit cac2d01

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

doc/unet.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,8 @@ The U-Net implementation can be found in the following GitHub repo:
9999

100100
* `Unet.py <https://github.com/Lasagne/Recipes/blob/master/modelzoo/Unet.py>`_ : Main script. Defines the model
101101

102+
* `train_unet.py <../code/unet/train_unet.py>`_ : Training loop.
103+
102104
The user can now build a U-Net with a specified number of input channels and number of classes.
103105

104106
First include the Lasagne layers needed to define the U-Net architecture :

0 commit comments

Comments
 (0)