Added Dockerfile to make it easier to use

Dockerfile

@@ -0,0 +1,18 @@
+FROM pytorch/pytorch:1.7.0-cuda11.0-cudnn8-runtime
+
+# Update system
+RUN apt update
+RUN apt install -y build-essential
+
+# Install Requirements
+COPY requirements.txt .
+RUN pip install --ignore-installed -r requirements.txt
+RUN pip install transformers
+RUN pip install --upgrade numpy
+
+# Copy over files and run
+COPY . .
+RUN python ./main_run.py --help
+
+# Enter and run what you like
+CMD /bin/bash

README.md

@@ -37,6 +37,21 @@ python main_run.py --folder=bgl/ --log_file=BGL.log --dataset_name=bgl --model_n
 python main_run.py --help
 ```
 
+
+### Docker
+
+#### Build
+
+```
+docker build -t lad .
+```
+
+#### Run
+
+```
+docker run --gpus all -it lad
+```
+
 ## Citation
 If you find the code and models useful for your research, please cite the following paper:
 ```