org.deeplearning4j.zoo.model.helper

Class NASNetHelper

java.lang.Object

org.deeplearning4j.zoo.model.helper.NASNetHelper

public class NASNetHelper
extends Object

Layer helpers NASNet.

Author:

Justin Long (crockpotveggies)

Constructor Summary

Constructors

Constructor and Description
NASNetHelper()

Method Summary

Modifier and Type	Method and Description
static String	adjustBlock(ComputationGraphConfiguration.GraphBuilder graphBuilder, int filters, String blockId, String input)
static String	adjustBlock(ComputationGraphConfiguration.GraphBuilder graphBuilder, int filters, String blockId, String input, String inputToMatch)
static org.nd4j.linalg.primitives.Pair<String,String>	normalA(ComputationGraphConfiguration.GraphBuilder graphBuilder, int filters, String blockId, String inputX, String inputP)
static org.nd4j.linalg.primitives.Pair<String,String>	reductionA(ComputationGraphConfiguration.GraphBuilder graphBuilder, int filters, String blockId, String inputX, String inputP)
static String	sepConvBlock(ComputationGraphConfiguration.GraphBuilder graphBuilder, int filters, int kernelSize, int stride, String blockId, String input)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

NASNetHelper

public NASNetHelper()

Method Detail

sepConvBlock

public static String sepConvBlock(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                  int filters,
                                  int kernelSize,
                                  int stride,
                                  String blockId,
                                  String input)

adjustBlock

public static String adjustBlock(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                 int filters,
                                 String blockId,
                                 String input)

adjustBlock

public static String adjustBlock(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                 int filters,
                                 String blockId,
                                 String input,
                                 String inputToMatch)

normalA

public static org.nd4j.linalg.primitives.Pair<String,String> normalA(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                                                     int filters,
                                                                     String blockId,
                                                                     String inputX,
                                                                     String inputP)

reductionA

public static org.nd4j.linalg.primitives.Pair<String,String> reductionA(ComputationGraphConfiguration.GraphBuilder graphBuilder,
                                                                        int filters,
                                                                        String blockId,
                                                                        String inputX,
                                                                        String inputP)