diff --git a/_ShA-base-train.sh b/_ShA-base-train.sh
index 0a4e8c5..90e3403 100644
--- a/_ShA-base-train.sh
+++ b/_ShA-base-train.sh
@@ -29,5 +29,6 @@ python train.py \
 	--save_path ./save/ShanghaiA-base-$(date -Iminutes) \
 	--batch_size 4 \
 	--gpus 0,1,2,3 \
-	--print_freq 100
+	--print_freq 100 \
+	--epochs 1000
 
diff --git a/_ShA-fb-train.sh b/_ShA-fb-train.sh
new file mode 100644
index 0000000..df6f26c
--- /dev/null
+++ b/_ShA-fb-train.sh
@@ -0,0 +1,35 @@
+#!/bin/sh
+#SBATCH -N 1
+#SBATCH -n 1
+#SBATCH --partition=Teach-Standard
+#SBATCH --gres=gpu:4
+#SBATCH --mem=24000
+#SBATCH --time=3-00:00:00
+
+set -e
+
+export CUDA_HOME=/opt/cuda-9.0.176.1/
+export CUDNN_HOME=/opt/cuDNN-7.0/
+export STUDENT_ID=$(whoami)
+
+export LD_LIBRARY_PATH=${CUDNN_HOME}/lib64:${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+export LIBRARY_PATH=${CUDNN_HOME}/lib64:${LIBRARY_PATH}
+export CPATH=${CUDNN_HOME}/include:$CPATH
+export PATH=${CUDA_HOME}/bin:${PATH}
+export PYTHON_PATH=$PATH
+
+mkdir -p /disk/scratch/${STUDENT_ID}
+export TMPDIR=/disk/scratch/${STUDENT_ID}/
+export TMP=/disk/scratch/${STUDENT_ID}/
+
+source /home/${STUDENT_ID}/miniconda3/bin/activate mlp-cuda
+
+python train.py \
+	--pth_tar './pretrained/deit-base-patch16-384/pytorch_model.bin' \
+	--model 'stn' \
+	--save_path ./save/ShanghaiA-fb-stn-$(date -Iminutes) \
+	--batch_size 4 \
+	--gpus 0,1,2,3 \
+	--print_freq 100 \
+	--epochs 1000
+
diff --git a/_ShA-train.sh b/_ShA-train.sh
index 0a6da04..31fa565 100644
--- a/_ShA-train.sh
+++ b/_ShA-train.sh
@@ -30,5 +30,6 @@ python train.py \
 	--save_path ./save/ShanghaiA-stn-$(date -Iminutes) \
 	--batch_size 4 \
 	--gpus 0,1,2,3 \
-	--print_freq 100
+	--print_freq 100 \
+	--epochs 1000
 
diff --git a/_ShB-base-train.sh b/_ShB-base-train.sh
index c4d0440..aeb7374 100644
--- a/_ShB-base-train.sh
+++ b/_ShB-base-train.sh
@@ -30,5 +30,5 @@ python train.py \
 	--save_path ./save/ShanghaiB-base-$(date -Iminutes) \
 	--batch_size 4 \
 	--gpus 0,1,2,3 \
-	--print_freq 100
-
+	--print_freq 100 \
+	--epochs 1000
diff --git a/_ShB-train.sh b/_ShB-train.sh
index 46fd063..dc15cc7 100644
--- a/_ShB-train.sh
+++ b/_ShB-train.sh
@@ -31,5 +31,5 @@ python train.py \
 	--save_path ./save/ShanghaiB-stn-$(date -Iminutes) \
 	--batch_size 4 \
 	--gpus 0,1,2,3 \
-	--print_freq 100
-
+	--print_freq 100 \
+	--epochs 1000