Upload Train_guide.ipynb

Train_script/Train_guide.ipynb

@@ -2,17 +2,17 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "id": "cd7ca034-54d2-434d-9ba8-9e1877e7a7c9",
    "metadata": {},
    "outputs": [],
    "source": [
-    "!unzip -q '/workspace/EfficientAT_code_train.zip' -d './train'"
+    "!unzip -q '/workspace/EfficientAT-main.zip' -d './train'"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "id": "2f674532-597d-4471-aef8-53c6f2b3ce7f",
    "metadata": {},
    "outputs": [],
@@ -27,12 +27,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "#Please change the dataset_config (filename hdf5) in \"audioset.py\" in \"EfficientAT-main/datasets\" "
+    "#Please change the dataset_config (filename hdf5) in \"audioset.py\" or \"emotion_dataset.py\" in \"EfficientAT-main/datasets\" "
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "id": "7dd14b84-3451-423f-bc07-16fddddc2a07",
    "metadata": {},
    "outputs": [
@@ -54,12 +54,12 @@
       "Requirement already satisfied: torchvision in /usr/local/lib/python3.10/dist-packages (from -r /workspace/train/EfficientAT-main/requirements.txt (line 8)) (0.16.0+cu118)\n",
       "Collecting tqdm (from -r /workspace/train/EfficientAT-main/requirements.txt (line 9))\n",
       "  Downloading tqdm-4.67.1-py3-none-any.whl.metadata (57 kB)\n",
-      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m57.7/57.7 kB\u001b[0m \u001b[31m3.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m57.7/57.7 kB\u001b[0m \u001b[31m2.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hCollecting wandb (from -r /workspace/train/EfficientAT-main/requirements.txt (line 10))\n",
       "  Downloading wandb-0.21.1-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (10 kB)\n",
       "Collecting pandas (from -r /workspace/train/EfficientAT-main/requirements.txt (line 11))\n",
       "  Downloading pandas-2.3.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (91 kB)\n",
-      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m91.2/91.2 kB\u001b[0m \u001b[31m2.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m91.2/91.2 kB\u001b[0m \u001b[31m21.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hCollecting seaborn (from -r /workspace/train/EfficientAT-main/requirements.txt (line 12))\n",
       "  Downloading seaborn-0.13.2-py3-none-any.whl.metadata (5.4 kB)\n",
       "Collecting audioread>=2.1.9 (from librosa->-r /workspace/train/EfficientAT-main/requirements.txt (line 3))\n",
@@ -68,7 +68,7 @@
       "  Downloading numba-0.61.2-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl.metadata (2.8 kB)\n",
       "Collecting scipy>=1.6.0 (from librosa->-r /workspace/train/EfficientAT-main/requirements.txt (line 3))\n",
       "  Downloading scipy-1.15.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (61 kB)\n",
-      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.0/62.0 kB\u001b[0m \u001b[31m2.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.0/62.0 kB\u001b[0m \u001b[31m7.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hCollecting joblib>=1.0 (from librosa->-r /workspace/train/EfficientAT-main/requirements.txt (line 3))\n",
       "  Downloading joblib-1.5.1-py3-none-any.whl.metadata (5.6 kB)\n",
       "Requirement already satisfied: decorator>=4.3.0 in /usr/local/lib/python3.10/dist-packages (from librosa->-r /workspace/train/EfficientAT-main/requirements.txt (line 3)) (5.1.1)\n",
@@ -103,7 +103,7 @@
       "  Downloading protobuf-6.32.0-cp39-abi3-manylinux2014_x86_64.whl.metadata (593 bytes)\n",
       "Collecting pydantic<3 (from wandb->-r /workspace/train/EfficientAT-main/requirements.txt (line 10))\n",
       "  Downloading pydantic-2.11.7-py3-none-any.whl.metadata (67 kB)\n",
-      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m68.0/68.0 kB\u001b[0m \u001b[31m2.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m68.0/68.0 kB\u001b[0m \u001b[31m15.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hRequirement already satisfied: pyyaml in /usr/local/lib/python3.10/dist-packages (from wandb->-r /workspace/train/EfficientAT-main/requirements.txt (line 10)) (6.0.1)\n",
       "Collecting sentry-sdk>=2.0.0 (from wandb->-r /workspace/train/EfficientAT-main/requirements.txt (line 10))\n",
       "  Downloading sentry_sdk-2.35.0-py2.py3-none-any.whl.metadata (10 kB)\n",
@@ -124,7 +124,7 @@
       "  Downloading cycler-0.12.1-py3-none-any.whl.metadata (3.8 kB)\n",
       "Collecting fonttools>=4.22.0 (from matplotlib!=3.6.1,>=3.4->seaborn->-r /workspace/train/EfficientAT-main/requirements.txt (line 12))\n",
       "  Downloading fonttools-4.59.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl.metadata (108 kB)\n",
-      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m108.9/108.9 kB\u001b[0m \u001b[31m5.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m108.9/108.9 kB\u001b[0m \u001b[31m39.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hCollecting kiwisolver>=1.3.1 (from matplotlib!=3.6.1,>=3.4->seaborn->-r /workspace/train/EfficientAT-main/requirements.txt (line 12))\n",
       "  Downloading kiwisolver-1.4.9-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl.metadata (6.3 kB)\n",
       "Requirement already satisfied: pyparsing>=2.3.1 in /usr/lib/python3/dist-packages (from matplotlib!=3.6.1,>=3.4->seaborn->-r /workspace/train/EfficientAT-main/requirements.txt (line 12)) (2.4.7)\n",
@@ -148,70 +148,70 @@
       "Collecting smmap<6,>=3.0.1 (from gitdb<5,>=4.0.1->gitpython!=3.1.29,>=1.0.0->wandb->-r /workspace/train/EfficientAT-main/requirements.txt (line 10))\n",
       "  Downloading smmap-5.0.2-py3-none-any.whl.metadata (4.3 kB)\n",
       "Downloading av-15.0.0-cp310-cp310-manylinux_2_28_x86_64.whl (39.2 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m39.2/39.2 MB\u001b[0m \u001b[31m46.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m39.2/39.2 MB\u001b[0m \u001b[31m110.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
       "\u001b[?25hDownloading h5py-3.14.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.6 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m4.6/4.6 MB\u001b[0m \u001b[31m31.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m4.6/4.6 MB\u001b[0m \u001b[31m50.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0mm\n",
       "\u001b[?25hDownloading librosa-0.11.0-py3-none-any.whl (260 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m260.7/260.7 kB\u001b[0m \u001b[31m60.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m260.7/260.7 kB\u001b[0m \u001b[31m86.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading scikit_learn-1.7.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (9.7 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m9.7/9.7 MB\u001b[0m \u001b[31m95.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0mta \u001b[36m0:00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m9.7/9.7 MB\u001b[0m \u001b[31m145.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0m\n",
       "\u001b[?25hDownloading tqdm-4.67.1-py3-none-any.whl (78 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m78.5/78.5 kB\u001b[0m \u001b[31m23.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m78.5/78.5 kB\u001b[0m \u001b[31m34.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading wandb-0.21.1-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (22.4 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m22.4/22.4 MB\u001b[0m \u001b[31m105.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m22.4/22.4 MB\u001b[0m \u001b[31m86.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n",
       "\u001b[?25hDownloading pandas-2.3.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (12.3 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m12.3/12.3 MB\u001b[0m \u001b[31m106.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m0:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m12.3/12.3 MB\u001b[0m \u001b[31m127.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0m\n",
       "\u001b[?25hDownloading seaborn-0.13.2-py3-none-any.whl (294 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m294.9/294.9 kB\u001b[0m \u001b[31m51.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m294.9/294.9 kB\u001b[0m \u001b[31m76.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading audioread-3.0.1-py3-none-any.whl (23 kB)\n",
       "Downloading click-8.2.1-py3-none-any.whl (102 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m102.2/102.2 kB\u001b[0m \u001b[31m31.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m102.2/102.2 kB\u001b[0m \u001b[31m37.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading gitpython-3.1.45-py3-none-any.whl (208 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m208.2/208.2 kB\u001b[0m \u001b[31m56.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m208.2/208.2 kB\u001b[0m \u001b[31m76.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading joblib-1.5.1-py3-none-any.whl (307 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m307.7/307.7 kB\u001b[0m \u001b[31m68.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m307.7/307.7 kB\u001b[0m \u001b[31m96.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading lazy_loader-0.4-py3-none-any.whl (12 kB)\n",
       "Downloading matplotlib-3.10.5-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (8.7 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m8.7/8.7 MB\u001b[0m \u001b[31m91.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m8.7/8.7 MB\u001b[0m \u001b[31m147.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0ma \u001b[36m0:00:01\u001b[0m\n",
       "\u001b[?25hDownloading msgpack-1.1.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (408 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m408.6/408.6 kB\u001b[0m \u001b[31m81.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m408.6/408.6 kB\u001b[0m \u001b[31m111.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading numba-0.61.2-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (3.8 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m3.8/3.8 MB\u001b[0m \u001b[31m138.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m3.8/3.8 MB\u001b[0m \u001b[31m97.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m\n",
       "\u001b[?25hDownloading pooch-1.8.2-py3-none-any.whl (64 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m64.6/64.6 kB\u001b[0m \u001b[31m3.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m64.6/64.6 kB\u001b[0m \u001b[31m1.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0meta \u001b[36m0:00:01\u001b[0m\n",
       "\u001b[?25hDownloading protobuf-6.32.0-cp39-abi3-manylinux2014_x86_64.whl (322 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m322.0/322.0 kB\u001b[0m \u001b[31m49.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m322.0/322.0 kB\u001b[0m \u001b[31m67.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading pydantic-2.11.7-py3-none-any.whl (444 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m444.8/444.8 kB\u001b[0m \u001b[31m82.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m444.8/444.8 kB\u001b[0m \u001b[31m105.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading pydantic_core-2.33.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (2.0 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m2.0/2.0 MB\u001b[0m \u001b[31m140.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m2.0/2.0 MB\u001b[0m \u001b[31m162.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading pytz-2025.2-py2.py3-none-any.whl (509 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m509.2/509.2 kB\u001b[0m \u001b[31m5.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m509.2/509.2 kB\u001b[0m \u001b[31m121.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading scipy-1.15.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (37.7 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m37.7/37.7 MB\u001b[0m \u001b[31m43.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m37.7/37.7 MB\u001b[0m \u001b[31m110.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0m\n",
       "\u001b[?25hDownloading sentry_sdk-2.35.0-py2.py3-none-any.whl (363 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m363.8/363.8 kB\u001b[0m \u001b[31m69.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━��━━━━━━━━━━━━━━\u001b[0m \u001b[32m363.8/363.8 kB\u001b[0m \u001b[31m101.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading soundfile-0.13.1-py2.py3-none-manylinux_2_28_x86_64.whl (1.3 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.3/1.3 MB\u001b[0m \u001b[31m128.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.3/1.3 MB\u001b[0m \u001b[31m153.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading soxr-0.5.0.post1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (252 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m252.8/252.8 kB\u001b[0m \u001b[31m44.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m252.8/252.8 kB\u001b[0m \u001b[31m86.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading threadpoolctl-3.6.0-py3-none-any.whl (18 kB)\n",
       "Downloading typing_extensions-4.14.1-py3-none-any.whl (43 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m43.9/43.9 kB\u001b[0m \u001b[31m11.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m43.9/43.9 kB\u001b[0m \u001b[31m16.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading tzdata-2025.2-py2.py3-none-any.whl (347 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m347.8/347.8 kB\u001b[0m \u001b[31m71.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m347.8/347.8 kB\u001b[0m \u001b[31m99.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading annotated_types-0.7.0-py3-none-any.whl (13 kB)\n",
       "Downloading contourpy-1.3.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (325 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m325.0/325.0 kB\u001b[0m \u001b[31m39.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m325.0/325.0 kB\u001b[0m \u001b[31m97.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading cycler-0.12.1-py3-none-any.whl (8.3 kB)\n",
       "Downloading fonttools-4.59.1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (4.8 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m4.8/4.8 MB\u001b[0m \u001b[31m113.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m4.8/4.8 MB\u001b[0m \u001b[31m129.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m\n",
       "\u001b[?25hDownloading gitdb-4.0.12-py3-none-any.whl (62 kB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.8/62.8 kB\u001b[0m \u001b[31m11.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.8/62.8 kB\u001b[0m \u001b[31m24.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading kiwisolver-1.4.9-cp310-cp310-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (1.6 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.6/1.6 MB\u001b[0m \u001b[31m72.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━��━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.6/1.6 MB\u001b[0m \u001b[31m150.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
       "\u001b[?25hDownloading llvmlite-0.44.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (42.4 MB)\n",
-      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m42.4/42.4 MB\u001b[0m \u001b[31m47.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m00:01\u001b[0m00:01\u001b[0mm\n",
+      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m42.4/42.4 MB\u001b[0m \u001b[31m83.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m:00:01\u001b[0m00:01\u001b[0m\n",
       "\u001b[?25hDownloading typing_inspection-0.4.1-py3-none-any.whl (14 kB)\n",
       "Downloading smmap-5.0.2-py3-none-any.whl (24 kB)\n",
       "Installing collected packages: pytz, tzdata, typing_extensions, tqdm, threadpoolctl, soxr, smmap, sentry-sdk, scipy, protobuf, msgpack, llvmlite, lazy_loader, kiwisolver, joblib, h5py, fonttools, cycler, contourpy, click, av, audioread, annotated-types, typing-inspection, soundfile, scikit_learn, pydantic-core, pooch, pandas, numba, matplotlib, gitdb, seaborn, pydantic, librosa, gitpython, wandb\n",
@@ -233,7 +233,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "id": "5a7a42e9-3f79-4f63-a6ea-98db2206ca96",
    "metadata": {},
    "outputs": [
@@ -262,19 +262,137 @@
    "execution_count": null,
    "id": "d0abf14b-46a1-4f92-9f70-e0de40dff46a",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Epoch 9/200: 100%|███████████| 145/145 [01:00<00:00,  2.38it/s, train_loss=1.09]\n",
+      "Validating: 100%|███████████████████████████████| 37/37 [00:08<00:00,  4.14it/s]\n",
+      "Confusion Matrix:\n",
+      "[[ 189    1    0  539  540   11    0]\n",
+      " [   2   13    0  129  711   27    0]\n",
+      " [   1    0    7   86  297   31    0]\n",
+      " [   4    0    0 1111  699    7    0]\n",
+      " [   1    0    0   67 2995    2    0]\n",
+      " [   4    0    0  257  982  218    0]\n",
+      " [   4    0    0   53  269    4    0]]\n",
+      "Epoch 9/200, Train Loss: 1.0852, Validation Loss: 1.8843, Validation Accuracy: 0.4895, LR: 0.001000, Time: 69.80s\n",
+      "Epoch 10/200:   0%|                                     | 0/145 [00:00<?, ?it/s]Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Setting temperature for attention over kernels to 21.0\n",
+      "Epoch 10/200: 100%|██████████| 145/145 [01:00<00:00,  2.38it/s, train_loss=1.05]\n",
+      "Validating: 100%|███████████████████████████████| 37/37 [00:08<00:00,  4.16it/s]\n",
+      "Confusion Matrix:\n",
+      "[[ 177    1    0  697  377   28    0]\n",
+      " [   2  176    0  130  553   21    0]\n",
+      " [   8    0    4   65  271   74    0]\n",
+      " [   3    1    0 1031  740   46    0]\n",
+      " [  10    0    0   63 2985    7    0]\n",
+      " [  18    0    0  171  883  389    0]\n",
+      " [  11   10    0   86  216    6    1]]\n",
+      "Epoch 10/200, Train Loss: 1.0541, Validation Loss: 1.9365, Validation Accuracy: 0.5143, LR: 0.001000, Time: 69.76s\n",
+      "Epoch 11/200:   0%|                                     | 0/145 [00:00<?, ?it/s]Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Setting temperature for attention over kernels to 20.0\n",
+      "Epoch 11/200:   1%|▏           | 2/145 [00:03<03:30,  1.47s/it, train_loss=2.47]"
+     ]
+    }
+   ],
    "source": [
     "!python ex_train_audio_classification_mode.py --cuda --train --pretrained --n_epochs=200 --model_name=dymn04_im --batch_size=256 --max_lr=0.001 --pretrain_final_temp=30 --adamw"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 27,
    "id": "d3b0a027-1774-4725-9e29-0112001369fe",
    "metadata": {},
    "outputs": [],
    "source": [
-    "!rm -rf '/workspace/train/EfficientAT-main/training_results_dymn04_im_20250816_032000'"
+    "!rm -rf '/workspace/train/EfficientAT-main/training_results_dymn04_im_20250816_072418'"
    ]
   },
   {