Integration with Comet.ml.
- Create account: comet.ml/signup.
- Export API key to the environment variable (more help here). In your terminal run:
export COMET_API_KEY='YOUR_LONG_API_TOKEN'
or include it in your ./comet.config
file (recommended). More help is here.
- You need to install neptune-client. In your terminal run:
pip install comet_ml
or (alternative installation using conda). In your terminal run:
conda install -c anaconda -c conda-forge -c comet_ml comet_ml
Key is to create the callback CometMLCallback
before you create Learner()
like this:
from fastai.callback.comet import CometMLCallback
comet_ml_callback = CometCallback('PROJECT_NAME') # specify project
learn = Learner(dls, model,
cbs=comet_ml_callback
)
learn.fit_one_cycle(1)