class BTNorm2d[source]

BTNorm2d(channels) :: Module

BatchTimeNorm2d module.

class LSTM[source]

LSTM(ni, nf) :: Module

LSTM module.

class SpaceConv[source]

SpaceConv(in_ch, out_ch, kernel_sz, stride) :: Module

SpaceConv module.

class UpSpaceConv[source]

UpSpaceConv(in_ch, out_ch) :: Module

UpSpaceConv module.

class TimeConv[source]

TimeConv(in_ch, out_ch, kernel_sz, stride, use_lstm) :: Module

TimeConv module.

class UpTimeConv[source]

UpTimeConv(in_ch, out_ch) :: Module

UpTimeConv module.

class SpaceTimeConv[source]

SpaceTimeConv(in_ch, out_ch, kernel_sz, time_sz, stride, time_stride, time_ch, use_lstm=False) :: Module

SpaceTimeConv module.

class UpSpaceTimeConv[source]

UpSpaceTimeConv(in_ch, out_ch, time_ch) :: Module

UpSpaceTimeConv module.

class BA_Net[source]

BA_Net(in_ch, n_classes, sequence_len) :: Module

BA-Net model.