modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer

Adapted from https://github.com/pytorch/pytorch/blob/master/torch/nn/modules/transformer.py

Classes

DiTDecoderLayer

previous

modelzoo.vision.pytorch.dit.layers.DiTDecoder.DiTDecoder

next

modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer.DiTDecoderLayer