Added acknowledgement.

aicenter · May 4, 2021 · 2c8f497 · 2c8f497
1 parent b43b255
commit 2c8f497
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/scripts/experiments_tabular/dagmm_run.sh b/scripts/experiments_tabular/dagmm_run.sh
@@ -1,7 +1,8 @@
 #!/bin/bash
+#SBATCH --partition=cpu
 #SBATCH --time=24:00:00
-#SBATCH --nodes=1 --ntasks-per-node=4 --cpus-per-task=1
-#SBATCH --mem=24G
+#SBATCH --nodes=1 --cpus-per-task=1
+#SBATCH --mem=16G
 #SBATCH --qos==collaborator
 
 MAX_SEED=$1

diff --git a/src/models/dagmm.jl b/src/models/dagmm.jl
@@ -1,3 +1,13 @@
+"""
+    Implements all needed parts for constructing DAGMM model
+        (https://sites.cs.ucsb.edu/~bzong/doc/iclr18-dagmm.pdf) for detecting anomalies.
+    Code is inspired by multiple reimplementations
+    - https://github.com/danieltan07/dagmm
+    - https://github.com/tnakae/DAGMM/
+    - https://github.com/Newcomer520/tf-dagmm
+    - https://github.com/mperezcarrasco/PyTorch-DAGMM
+"""
+
 using Flux: mse, softmax, unsqueeze, stack
 using LinearAlgebra
 using StatsBase