modify e2e tests for asgd training

This commit is contained in:
Qiwei Ye 2016-10-18 16:04:59 +08:00
Родитель eecce32487
Коммит cc40f924b8
6 изменённых файлов: 18 добавлений и 15 удалений

Двоичный файл не отображается.

Просмотреть файл

@ -86,7 +86,7 @@ Train = [
reader = [
readerType = "ImageReader"
file = "$DataDir$/cifar-10-batches-py/train_map.txt"
file = "$DataDir$/train_map.txt"
randomize = "auto"
features = [
width = 32
@ -96,7 +96,7 @@ Train = [
cropRatio = 0.8
jitterType = "uniRatio"
interpolations = "linear"
meanFile = "$DataDir$/cifar-10-batches-py/CIFAR-10_mean.xml"
meanFile = "$DataDir$/CIFAR-10_mean.xml"
]
labels = [
labelDim = 10
@ -105,7 +105,7 @@ Train = [
cvReader = [
readerType = "ImageReader"
file = "$DataDir$/cifar-10-batches-py/test_map.txt"
file = "$DataDir$/test_map.txt"
randomize = "none"
features = [
width = 32
@ -115,7 +115,7 @@ Train = [
cropRatio = 1
jitterType = "uniRatio"
interpolations = "linear"
meanFile = "$DataDir$/cifar-10-batches-py/CIFAR-10_mean.xml"
meanFile = "$DataDir$/CIFAR-10_mean.xml"
]
labels = [
labelDim = 10

Просмотреть файл

Просмотреть файл

@ -2,20 +2,21 @@
. $TEST_DIR/run-test-common
#dataDir="."
ConfigDir=$TEST_DIR
LogFileName=stderr
LogFileName="ASGDMultiGPU"
Instances=4
NumCPUThreads=$(threadsPerInstance $Instances)
# cntkmpirun <MPI args> <CNTK config file name> <additional CNTK args>
cntkmpirun "-n $Instances" 03_ResNet-parallel.cntk "numCPUThreads=$NumCPUThreads precision=float SimpleMultiGPU=[SGD=[ParallelTrain=[DataParallelSGD=[gradientBits=32]]]]"
cntkmpirun "-n $Instances" 03_ResNet-parallel.cntk "numCPUThreads=$NumCPUThreads precision=float parallelTrain=true SimpleMultiGPU=[SGD=[ParallelTrain=[DataParallelSGD=[gradientBits=32]]]]"
ExitCode=$?
sed 's/^/MPI Rank 0: /' $TEST_RUN_DIR/"$LogFileName"_ASGDMultiGPU.logrank0
sed 's/^/MPI Rank 1: /' $TEST_RUN_DIR/"$LogFileName"_ASGDMultiGPU.logrank1
sed 's/^/MPI Rank 2: /' $TEST_RUN_DIR/"$LogFileName"_ASGDMultiGPU.logrank2
sed 's/^/MPI Rank 3: /' $TEST_RUN_DIR/"$LogFileName"_ASGDMultiGPU.logrank3
sed 's/^/MPI Rank 0: /' $TEST_RUN_DIR/"$LogFileName"_Train.logrank0
sed 's/^/MPI Rank 1: /' $TEST_RUN_DIR/"$LogFileName"_Train.logrank1
sed 's/^/MPI Rank 2: /' $TEST_RUN_DIR/"$LogFileName"_Train.logrank2
sed 's/^/MPI Rank 3: /' $TEST_RUN_DIR/"$LogFileName"_Train.logrank3
# Delete the test data if copied
[[ "$Copied" -eq "1" ]] && rm -rf "$DataDir"
#[[ "$Copied" -eq "1" ]] && rm -rf "$DataDir"
exit $ExitCode

Просмотреть файл

@ -4,6 +4,7 @@
ConfigDir=$TEST_DIR
if [[ ! -d $TEST_DATA_DIR || ! -e $TEST_DATA_DIR/Train_cntk_text.txt || ! -e $TEST_DATA_DIR/train_map.txt ]]; then
echo "test_data_dir";
# Cannot find test data locally.
# Try external test data directory (not part of the CNTK repository) as an alternative.
if [[ -d "$CNTK_EXTERNAL_TESTDATA_SOURCE_DIRECTORY" ]]; then
@ -16,9 +17,9 @@ if [[ ! -d $TEST_DATA_DIR || ! -e $TEST_DATA_DIR/Train_cntk_text.txt || ! -e $TE
# Copy the test data to the test run directory
DataDir=$TEST_RUN_DIR/TestData
mkdir $DataDir
mkdir $DataDir/cifar-10-batches-py
mkdir $DataDir/cifar-10-batches-py
cp -R $DataSourceDir/*_cntk_text.txt $DataDir || exit $?
cp -R $DataSourceDir/cifar-10-batches-py/data.zip $DataDir/cifar-10-batches-py || exit $?
#cp -R $DataSourceDir/cifar-10-batches-py/data.zip $DataDir/cifar-10-batches-py || exit $?
cp -R $DataSourceDir/cifar-10-batches-py/CIFAR-10_mean.xml $DataDir || exit $?
cp -R $DataSourceDir/cifar-10-batches-py/*_map.txt $DataDir || exit $?
Copied=1
@ -26,4 +27,4 @@ if [[ ! -d $TEST_DATA_DIR || ! -e $TEST_DATA_DIR/Train_cntk_text.txt || ! -e $TE
echo Error: cannot find data. Please see Examples/Image/DataSets/CIFAR10/README.md for instructions to get it.
exit 1
fi
fi
fi

Просмотреть файл

@ -1,4 +1,5 @@
# dataDir: ../../Data
dataDir: .
#../../Data
tags:
# running on every BVT job in 'P' (Parallel) leg in Debug-GPU and Release-CPU configurations: