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

Method add_train_ops

caffe2/python/layers/layers.py:464–467  ·  view source on GitHub ↗
(self, net)

Source from the content-addressed store, hash-verified

462 self.add_ops(net)
463
464 def add_train_ops(self, net):
465 # Default train layer implementation is completely matching
466 # eval layer implementation.
467 self.add_eval_ops(net)
468
469 def add_ops_to_accumulate_pred(self, net):
470 # This adds operators to accumulate predictions/labels/weights. The

Callers 1

add_operatorsMethod · 0.95

Calls 1

add_eval_opsMethod · 0.95

Tested by

no test coverage detected