MCPcopy
hub / github.com/pytorch/pytorch / iter

Function iter

caffe2/python/helpers/train.py:22–34  ·  view source on GitHub ↗
(model, blob_out, **kwargs)

Source from the content-addressed store, hash-verified

20
21
22def iter(model, blob_out, **kwargs):
23 if 'device_option' in kwargs:
24 del kwargs['device_option']
25 model.param_init_net.ConstantFill(
26 [],
27 blob_out,
28 shape=[1],
29 value=0,
30 dtype=core.DataType.INT64,
31 device_option=core.DeviceOption(caffe2_pb2.CPU, 0),
32 **kwargs
33 )
34 return model.net.Iter(blob_out, blob_out, **kwargs)
35
36
37def accuracy(model, blob_in, blob_out, **kwargs):

Callers 15

__iter__Method · 0.85
__iter__Method · 0.85
__iter__Method · 0.85
__iter__Method · 0.85
__iter__Method · 0.85
__iter__Method · 0.85
__iter__Method · 0.85
_accumulateFunction · 0.85
_reorder_tensors_asFunction · 0.85
visualize_graph_executorFunction · 0.85
build_extensionsMethod · 0.85
__init__Method · 0.85

Calls 1

IterMethod · 0.80

Used in the wild real call sites across dependent graphs

searching dependent graphs…