Add support for jupyter notebooks for training

scripts/bootstrap.sh

@@ -18,6 +18,11 @@ if [ -z "$SIA_REPO_PAT" ]; then
     exit 1
 fi
 
+# Install required packages
+apt-get update
+apt-get install -y \
+    vim
+
 # Create directory structure
 echo "Creating directory structure..."
 mkdir -p "/root/data/iterations"
@@ -33,6 +38,7 @@ if [ ! -d "/root/sia" ]; then
     cd "/root/sia"
     git config --global user.name "Niels Geens"
     git config --global user.email "niels.geens@gmail.com"
+    git config --global core.editor vim
 fi
 
 # Fixing permissions
@@ -58,10 +64,10 @@ ln -s "/root/sia/web/dist" "/root/static"
 # Install SIA dependencies
 source "/root/sia/scripts/install.sh"
 
-# Finetune model
-echo "Finetuning model..."
-train
-
-# Start SIA using restart script
-echo "Run restart script..."
-"/root/sia/scripts/restart.sh"
+## Finetune model
+#echo "Finetuning model..."
+#train
+#
+## Start SIA using restart script
+#echo "Run restart script..."
+#"/root/sia/scripts/restart.sh"

scripts/install.sh

@@ -9,6 +9,7 @@ source "/etc/profile.d/venv_itb.sh"
 echo "Installing Train tool..."
 python3 -m venv "/root/venvs/train"
 /root/venvs/train/bin/pip install -e /root/sia/tools/train/
+/root/venvs/train/bin/ipython kernel install --name=train
 echo "PATH=\"/root/venvs/train/bin/:\$PATH\"" > "/etc/profile.d/venv_train.sh"
 source "/etc/profile.d/venv_train.sh"