From 557a7c589a497efaa56f9d3cf12328df3a1903f7 Mon Sep 17 00:00:00 2001
From: Vithursan Thangarasa <vithursan.thangarasa@gmail.com>
Date: Thu, 8 Mar 2018 22:44:56 -0500
Subject: [PATCH] Remove Docker GPU training instructions. Needs to be tested
 with nvidia-docker.

---
 README.md | 18 +-----------------
 1 file changed, 1 insertion(+), 17 deletions(-)

diff --git a/README.md b/README.md
index 8fb7530..3d58540 100644
--- a/README.md
+++ b/README.md
@@ -52,23 +52,7 @@ Train ConvNet with Magnet Loss on the local machine using MNIST dataset:
 python magnet_loss_test.py --lr 1e-4 --batch-size 64 --mnist --dml
 ```
 
-## Docker
-
-### Docker: Installation
-
-**Prerequisites: Docker installed on your machine. If you don't have Docker installed already, then go here to [Docker Setup](https://docs.docker.com/engine/getstarted/step_one/)**
-
-To build Docker image:
-
-```sh
-docker build -t magnetloss:latest .
-```
-
-### Docker: Train
-To deploy and train on Docker container:
-```sh
-docker run -it magnetloss:latest python --lr 1e-4 --batch-size 64 --mnist --dml
-```
+## Docker GPU Training (WIP)
 
 ## Results
 ### MNIST