spyrit.external.drunet.UNetRes

class spyrit.external.drunet.UNetRes(in_nc=1, out_nc=1, nc=[64, 128, 256, 512], nb=4, act_mode='R', downsample_mode='strideconv', upsample_mode='convtranspose')

Bases: Module

Methods

forward(x0)

Define the computation performed at every call.