i’m trying the mobile vit transformers model and everything worked perfectly till the last part where i had to download going_modular module from git then take it to the working directory bc the code doesn’t work without it apparently, my question : is the problem “from going_modular import engine” or something else, and is there an alternative ?
literally nothing appears ,the notebook is stuck basically
maybe the data is the problem ?

from going_modular.going_modular import engine

# Create optimizer and loss function
optimizer = torch.optim.Adam(params=pretrained_vit.parameters(), 
                             lr=1e-3)
loss_fn = torch.nn.CrossEntropyLoss()

# Train the classifier head of the pretrained ViT feature extractor model

pretrained_vit_results = engine.train(model=pretrained_vit,
                                      train_dataloader=train_dataloader_pretrained,
                                      test_dataloader=test_dataloader_pretrained,
                                      optimizer=optimizer,
                                      loss_fn=loss_fn,
                                      epochs=10,
                                      device=device)

fzefzefzefzfzefzefzefzeefzefzef