From b86bf9b10898a1e97ca0e66ba473d0b449f04182 Mon Sep 17 00:00:00 2001
From: Kingslayer <filippo.gatti@centralesupelec.fr>
Date: Mon, 20 Jul 2020 16:50:20 +0200
Subject: [PATCH] [ADD] sh for bb

---
 README.tex.md | 4 ++--
 run-gpu-bb.sh | 5 +++++
 2 files changed, 7 insertions(+), 2 deletions(-)
 create mode 100755 run-gpu-bb.sh

diff --git a/README.tex.md b/README.tex.md
index f6d84d3..cc0c36c 100644
--- a/README.tex.md
+++ b/README.tex.md
@@ -54,7 +54,7 @@ Each action implies the choice of a corresponding `strategy` (to be specified in
 Extra keywords can be added as column's headers in the `strategy.txt` file: they are needed for comparison purposes and/or to test the discriminator performances.
 ### Signal Databases
 
-To train/test the different `DCAAE`, an extraction of 5000 signals from the [STEAD database](https://github.com/smousavi05/STEAD/) is provided in the `database` folder. Seismic signals are 40.96 s-long, sampled at 100 Hz (`nt`=4096 time steps).
+To train/test the different `DCAAE`, an extraction of 100 signals from the [STEAD database](https://github.com/smousavi05/STEAD/) is provided in the `database` folder. Seismic signals are 40.96 s-long, sampled at 100 Hz (`nt`=4096 time steps).
 
  - `ths_trn_nt4096_ls128_nzf8_nzd32.pth`: training set (80%)
  - `ths_tst_nt4096_ls128_nzf8_nzd32.pth`: testing set  (10%)
@@ -65,7 +65,7 @@ To train/test the different `DCAAE`, an extraction of 5000 signals from the [STE
 `nzf`: latent space channels (`filtered`)
 
 <p align="center">
-  <img src="MRD_eqk_scatter.png" width="350" height="350" title="Figure 1: Hypocentral distance $R_{hyp}$, magnitude $M_W$ and depth distribution of the earthquake sources">
+  <img src="MRD_eqk_scatter.png" width="350" height="233" title="Figure 1: Hypocentral distance, magnitude and depth distribution of the earthquake sources">
 </p>
 
 ## 
diff --git a/run-gpu-bb.sh b/run-gpu-bb.sh
new file mode 100755
index 0000000..94543a5
--- /dev/null
+++ b/run-gpu-bb.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+export PYTHONPATH="./src"
+
+python3 ./src/aae_drive_bbfl.py --dataroot='./database/stead' --dataset='nt4096_ls128_nzf8_nzd32.pth' --cutoff=1. --imageSize=4096 --latentSize=128  --niter=1 --cuda --ngpu=1 --nzd=32 --rlr=0.0001 --glr=0.0001 --outf='./imgs' --workers=8 --nsy=5000 --batchSize=100 --actions='./actions_bb_plt.txt' --strategy='./strategy_bb_plt.txt' 
+