Zoo model TF import test updates (#75)

* argLine fix, update compression_gru comment

* updated comment for xception

* undid but commented argLine change

* updated xlnet comment

* copyright headers
master
Ryan Nett 2019-07-22 12:58:25 -07:00 committed by AlexDBlack
parent b9708be5db
commit 00e6296140
4 changed files with 37 additions and 3 deletions

View File

@ -193,6 +193,8 @@
<!--
Maximum heap size was set to 8g, as a minimum required value for tests run.
Depending on a build machine, default value is not always enough.
For testing large zoo models, this may not be enough (so comment it out).
-->
<argLine>-Ddtype=float -Xmx8g</argLine>
</configuration>

View File

@ -56,13 +56,13 @@ public class TFGraphTestZooModels { //Note: Can't extend BaseNd4jTest here as we
public static final String[] IGNORE_REGEXES = {
//2019/07/10 - Libnd4j assign error - https://github.com/eclipse/deeplearning4j/issues/8002
//2019/07/22 - Result value failure
"xlnet_cased_L-24_H-1024_A-16",
//2019/07/03 - Out of Memory error
// 2019/07/22 - OOM, Passes with sufficient memory (16GB heap, 32GB off-heap tested)
"compression_residual_gru",
//2019/07/03 - Out of Memory error
// 2019/07/22 - OOM, Passes with sufficient memory (16GB heap, 32GB off-heap tested)
"deeplabv3_xception_ade20k_train",
//2019/07/03 - o.n.i.g.t.TFGraphMapper - No TensorFlow descriptor found for tensor "sample_sequence/model/h0/attn/MatMul", op "BatchMatMulV2"

View File

@ -1,3 +1,19 @@
/*
* Copyright (c) 2015-2019 Skymind, Inc.
*
* This program and the accompanying materials are made available under the
* terms of the Apache License, Version 2.0 which is available at
* https://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*
* SPDX-License-Identifier: Apache-2.0
*/
package org.nd4j.imports.listeners;
import lombok.NonNull;

View File

@ -1,3 +1,19 @@
/*
* Copyright (c) 2015-2019 Skymind, Inc.
*
* This program and the accompanying materials are made available under the
* terms of the Apache License, Version 2.0 which is available at
* https://www.apache.org/licenses/LICENSE-2.0.
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*
* SPDX-License-Identifier: Apache-2.0
*/
package org.nd4j.imports.listeners;
import org.apache.commons.io.FileUtils;