Set idx_embedding=0 as default. This way, class number > 1 works.