Attention56Bone

class torchelie.models.Attention56Bone(num_classes: int)

Attention56 bone

Parameters

in_ch (int) – number of channels in the images

training: bool