Open gursi26 opened 5 months ago
Hey @gursi26, Aim provides a built-in callback manager specifically designed for PyTorch Lightning. It is highly recommended to utilize this manager to handle all PyTorch Lightning events effectively.
Here's how your code snippet will appear after making those changes:
import lightning as L
aim_logger = AimLogger(
experiment_name="trial_exp_lightning",
)
trainer = L.Trainer(
max_epochs=EPOCHS,
accelerator=DEVICE,
logger = aim_logger
)
trainer.fit(training_module, train_loader, test_loader)
You can refer for more details on this page: https://aimstack.readthedocs.io/en/latest/quick_start/integrations.html#integration-with-pytorch-lightning
Feel free to share any additional issues you may encounter ☺️
❓Question
Here is my code for the PyTorch lightning trainer
Upon running this, I get the following error message.
I have checked previous issues in both the lightning and aim repos and it seems that integration is possible, so im wondering how I should go about fixing this.