seisnn.model.attention.transformer

seisnn.model.attention.transformer(img_rows=None, img_cols=None, color_type=3, num_class=3)

EQtransformer_model

Parameters

• img_rows – Height of the data.

• img_cols – Width of the data.

• color_type – Channel number of the data.

• num_class – Output class number.

Returns

EQtransformer model.