-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathREADME
38 lines (25 loc) · 1.15 KB
/
README
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
################################################################
Please read the following instruction before you training models:
You can train models in following ways:
./aistation_DL_train.sh tensorflow mnist 1
which means training tensorflow model with mnist by using 1 GPU
You can use more GPUS like this:
./aistation_DL_train.sh tensorflow mnist 2
Also you can train the models with MPI by this:
./aistation_DL_train.sh tensorflow mnist mpi 2
Please Note:
the default user is 'inspur', so please modify the
path which is used in training scripts if you want
to test those scripts in your own directory.
Or use the user_path_change.sh to new user like:
./user_path_change.sh jhon
Now we support following models:
-------------------------------
tensorflow mnist(cifar10)
pytorch mnist(cifar10)
caffe mnist(cifar10)
mxnet mnist
paddle mnist
------------------------------
in one or more GPUs, or in MPI (only mnist) training mode.
#############################################################