Update cnntrain_test.m