attention56

class torchelie.models.attention56(num_classes)

Experimental: Build a attention56 network

Parameters

• num_classes (int) – number of classes

• in_ch (int) – number of channels in the images

Warning

attention56() is experimental, and may change or be deleted soon if not already broken