feat/tf_support #37
|
@ -29,7 +29,7 @@ from aiia.model import AIIAConfig
|
||||||
from aiia.pretrain import Pretrainer
|
from aiia.pretrain import Pretrainer
|
||||||
|
|
||||||
# Create your model
|
# Create your model
|
||||||
config = AIIAConfig(model_name="AIIA-Base-512x20k")
|
config = AIIAConfig(model_type="AIIA-Base-512x20k")
|
||||||
model = AIIABase(config)
|
model = AIIABase(config)
|
||||||
|
|
||||||
# Initialize pretrainer with the model
|
# Initialize pretrainer with the model
|
||||||
|
|
Loading…
Reference in New Issue