atom-predict/egnn_v2/model/label.txt

cz:  
class_dict = {
    0: 'atom',
    1: 'cz',
}

xj:
class_dict = {
    0: '0',
    1: '1',
    2: '2',
}

smov:
class_dict = {
        0: 'S',
        1: 'Mo',
        2: 'V'
    }