search

LYH-YF / MWPToolkit

MWPToolkit is an open-source framework for math word problem(MWP) solvers.
MIT License
163 stars 37 forks source link

About "robertagen can't be found" #16

Open Changgeng-Wei opened 2 years ago

Changgeng-Wei commented 2 years ago

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 commented 2 years ago

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.