Skip to content

Instantly share code, notes, and snippets.

@jcreinhold

jcreinhold/model.py

Created January 3, 2019 15:24
Show Gist options
  • Save jcreinhold/afa7edbcf240e1bc024785381d621b3f to your computer and use it in GitHub Desktop.
Save jcreinhold/afa7edbcf240e1bc024785381d621b3f to your computer and use it in GitHub Desktop.
Download ZIP
define basic resnet
Raw
model.py
norm = 'batch'
layers = ([res3d_block(1,15,7,norm=norm,dense=True)] +
[res3d_block(16,16,norm=norm) for _ in range(4)] +
[conv3d(16,1,ks=1,pad=0,norm=None)])
model = nn.Sequential(*layers)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment