From c00b07f44fddb198b824e854ee838cc2db333fb6 Mon Sep 17 00:00:00 2001
From: Cassandra Grzonkowski <c_grzonkow18@cs.uni-kl.de>
Date: Thu, 25 Jan 2024 05:28:39 +0100
Subject: [PATCH] update load model on cluster

---
 main.py          | 4 ----
 model_run.sbatch | 1 -
 preprocess.py    | 5 -----
 requirements.txt | 2 +-
 4 files changed, 1 insertion(+), 11 deletions(-)

diff --git a/main.py b/main.py
index da33abe..15bc2c5 100644
--- a/main.py
+++ b/main.py
@@ -18,14 +18,10 @@ import torch.nn.functional as F
 import torchvision.transforms as transforms
 import numpy as np
 import matplotlib.pyplot as plt
-import IPython.display as display
 import librosa
 import librosa.display
-import PIL.Image
 from models.transformer import *
 from SongPacksDataset import *
-from tqdm.notebook import tqdm
-from sklearn.neural_network import MLPRegressor
 from preliminary_experiment import *
 from preprocess import *
 from torch import nn, Tensor
diff --git a/model_run.sbatch b/model_run.sbatch
index 940b94e..2700dc8 100644
--- a/model_run.sbatch
+++ b/model_run.sbatch
@@ -16,7 +16,6 @@ module load anaconda3/latest
 conda activate danceformer
 #pip install -r requirements.txt
 #pip install simfile
-pip install torchvision
 #python main.py --vocabulary /scratch/grzonkow/vocabulary --processes $SLURM_NTASKS --threads $SLURM_CPUS_PER_TASK "$@"
 python main.py --processes $SLURM_NTASKS --threads $SLURM_CPUS_PER_TASK "$@"
 conda deactivate
diff --git a/preprocess.py b/preprocess.py
index 15c20a7..cce6ad2 100644
--- a/preprocess.py
+++ b/preprocess.py
@@ -14,17 +14,12 @@ import torch.nn.functional as F
 import torchvision.transforms as transforms
 import numpy as np
 import matplotlib.pyplot as plt
-import IPython.display as display
 import librosa
 import librosa.display
-import PIL.Image
 from models.transformer import *
 from SongPacksDataset import *
-from tqdm.notebook import tqdm
-from sklearn.neural_network import MLPRegressor
 from preliminary_experiment import *
 from simfile.timing.engine import TimingEngine
-# from chart_embedding import *
 from torch import nn, Tensor
 from torch.nn import TransformerEncoder, TransformerEncoderLayer
 import re
diff --git a/requirements.txt b/requirements.txt
index 48aeca5..52350a7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,4 @@ torchaudio~=2.0.2
 torch~=2.0.1
 matplotlib
 librosa
-torchvision
\ No newline at end of file
+torchvision
-- 
GitLab