AIIA

Example Usage:

if __name__ == "__main__":
    data_path1 = "/root/training_data/vision-dataset/images_checkpoint.parquet"
    data_path2 = "/root/training_data/vision-dataset/vec_images_dataset.parquet"
    
    from aiia.model import AIIABase
    from aiia.model.config import AIIAConfig
    from aiia.pretrain import Pretrainer
    
    config = AIIAConfig(model_name="AIIA-Base-512x20k")
    model = AIIABase(config)
    
    pretrainer = Pretrainer(model, learning_rate=1e-4)
    pretrainer.train(data_path1, data_path2, num_epochs=10)