megatron.model.classification#

Description

Classification model.

Classes

Classification(num_classes[, ...])