About "robertagen can't be found"

[Changgeng-Wei]

Hi, I wanted to trained the model by RobertaGen, but it always showed the error "robertagen can't be found", why did it happend? That's my code below: python run_mwptoolkit.py --model=RoBERTaGen --dataset=math23k --task_type=single_equation --equation_fix=prefix --test_step=10 --gpu_id=0

[LYH-YF]

we name the model class as RobertaGen, not RoBERTaGen,so you setup --model=RobertaGen to find the model.

the framework is not supporting finding model through different uppercase or lowercase letters yet.