WIP Move OpenAI examples to their own folder

comet-ml · Nov 23, 2023 · c41593c · c41593c
1 parent c09a992
commit c41593c
Show file tree

Hide file tree

Showing 7 changed files with 677 additions and 776 deletions.
diff --git a/fastai/README.md b/fastai/README.md
@@ -0,0 +1 @@
+All of the fastai examples has been moved here: https://github.com/comet-ml/comet-examples/tree/master/integrations/model-training/fastai/.
diff --git a/fastai/train-example.py b/fastai/train-example.py
diff --git a/integrations/model-training/fastai/fastai-hello-world/README.md b/integrations/model-training/fastai/fastai-hello-world/README.md
diff --git a/integrations/model-training/fastai/fastai-hello-world/fastai_hello_world.py b/integrations/model-training/fastai/fastai-hello-world/fastai_hello_world.py
@@ -0,0 +1,40 @@
+# coding: utf-8
+import comet_ml
+
+from fastai.vision.all import (
+    Categorize,
+    Datasets,
+    GrandparentSplitter,
+    IntToFloatTensor,
+    PILImageBW,
+    ToTensor,
+    URLs,
+    error_rate,
+    get_image_files,
+    parent_label,
+    resnet18,
+    untar_data,
+    vision_learner,
+)
+
+EPOCHS = 5
+
+comet_ml.init(project_name="comet-examples-fastai-hello-world")
+experiment = comet_ml.Experiment()
+
+path = untar_data(URLs.MNIST_TINY)
+
+items = get_image_files(path)
+tds = Datasets(
+    items,
+    [PILImageBW.create, [parent_label, Categorize()]],
+    splits=GrandparentSplitter()(items),
+)
+dls = tds.dataloaders(after_item=[ToTensor(), IntToFloatTensor()])
+
+learn = vision_learner(dls, resnet18, pretrained=True, metrics=error_rate)
+
+with experiment.train():
+    learn.fit_one_cycle(EPOCHS)
+
+experiment.end()
diff --git a/integrations/model-training/fastai/fastai-hello-world/requirements.txt b/integrations/model-training/fastai/fastai-hello-world/requirements.txt
@@ -0,0 +1,2 @@
+comet_ml
+fastai
diff --git a/integrations/model-training/fastai/notebooks/fastai_hello_world.ipynb b/integrations/model-training/fastai/notebooks/fastai_hello_world.ipynb