MCPcopy Create free account
hub / github.com/pytorch/pytorch / get_predict_net

Method get_predict_net

caffe2/python/layer_test_util.py:78–79  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

76 return layer_model_instantiator.generate_eval_net(self.model)
77
78 def get_predict_net(self):
79 return layer_model_instantiator.generate_predict_net(self.model)
80
81 def run_train_net(self):
82 self.model.output_schema = schema.Struct()

Callers 15

testSparseDropoutWithReplacementMethod · 0.80
testFCWithoutBiasMethod · 0.80
testSparseLookupSumPoolingMethod · 0.80
testSparseLookupPositionWeightedOnIdListMethod · 0.80
testSparseLookupPositionWeightedOnIdScoreListMethod · 0.80
testSparseLookupRecencyWeightedOnIdScoreListMethod · 0.80
testSamplingTrainMethod · 0.80
testBatchNormalizationMethod · 0.80
testMapToRangeMethod · 0.80
testDropoutMethod · 0.80
testRandomFourierFeaturesMethod · 0.80
testArcCosineFeatureMapMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected