Add login to Comet if needed

comet-ml · Jul 3, 2024 · 537c8b2 · 537c8b2
1 parent 87ec6d3
commit 537c8b2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/integrations/model-training/deepspeed/deepspeed-cifar/cifar10_deepspeed.py b/integrations/model-training/deepspeed/deepspeed-cifar/cifar10_deepspeed.py
@@ -1,6 +1,8 @@
 # coding: utf-8
 import argparse
 
+import comet_ml
+
 import deepspeed
 import torch
 import torch.nn as nn
@@ -10,6 +12,9 @@
 from deepspeed.accelerator import get_accelerator
 from deepspeed.moe.utils import split_params_into_different_moe_groups_for_optimizer
 
+# Login to Comet if needed
+comet_ml.init()
+
 
 def add_argument():
     parser = argparse.ArgumentParser(description="CIFAR")