diff options
Diffstat (limited to 'tests/training/restarting/test_sgd_for_two_epochs_sync.sh')
-rw-r--r-- | tests/training/restarting/test_sgd_for_two_epochs_sync.sh | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/training/restarting/test_sgd_for_two_epochs_sync.sh b/tests/training/restarting/test_sgd_for_two_epochs_sync.sh index 4808894..d3ee295 100644 --- a/tests/training/restarting/test_sgd_for_two_epochs_sync.sh +++ b/tests/training/restarting/test_sgd_for_two_epochs_sync.sh @@ -8,6 +8,8 @@ rm -rf sgd_sync_2e sgd_sync_*_epoch.log mkdir -p sgd_sync_2e extra_opts="--no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 --optimizer sgd --sync-sgd" +# Added because default options has changes +extra_opts="$extra_opts --cost-type ce-mean --disp-label-counts false" # Uncomment to prepare the expected output |