Skip to content

Commit

Permalink
Remove Docker GPU training instructions. Needs to be tested with nvid…
Browse files Browse the repository at this point in the history
…ia-docker.
  • Loading branch information
vithursant committed Mar 9, 2018
1 parent 0a221e2 commit 557a7c5
Showing 1 changed file with 1 addition and 17 deletions.
18 changes: 1 addition & 17 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 557a7c5

Please sign in to comment.