seisnn.model.attention

Functions

ResNet_build(filter_nums, block_nums[, strides])

Build ResNet

transformer([img_rows, img_cols, …])

EQtransformer_model

Classes

MultiHeadSelfAttention(*args, **kwargs)

ResBlock(*args, **kwargs)

Residual CNN block Structure

TransformerBlockD(*args, **kwargs)

TransformerBlockE(*args, **kwargs)