Spaces:

teticio
/

audio-diffusion

Running

nturusin commited on Dec 28, 2022

Commit

5a187d9

•

1 Parent(s): 65031fe

Changed 'scripts/train_unconditional.py' to 'scripts/train_unet.py' in training

Files changed (1) hide show

notebooks/train_model.ipynb CHANGED Viewed

@@ -133,7 +133,7 @@
    },
    "outputs": [],
    "source": [
-    "!python scripts/train_unconditional.py \\\n",
     "  --dataset_name data \\\n",
     "  --output_dir model \\\n",
     "  --num_epochs 10 \\\n",

    },
    "outputs": [],
    "source": [
+    "!python scripts/train_unet.py \\\n",
     "  --dataset_name data \\\n",
     "  --output_dir model \\\n",
     "  --num_epochs 10 \\\n",