modify e2e tests for asgd training
This commit is contained in:
Родитель
eecce32487
Коммит
cc40f924b8
Двоичные данные
Tests/EndToEndTests/ParallelTraining/AsynchronizedSGD/.03_ResNet-parallel.cntk.swp
Normal file
Двоичные данные
Tests/EndToEndTests/ParallelTraining/AsynchronizedSGD/.03_ResNet-parallel.cntk.swp
Normal file
Двоичный файл не отображается.
|
@ -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:
|
||||
|
|
Загрузка…
Ссылка в новой задаче