(model, blob_out, **kwargs)
| 20 | |
| 21 | |
| 22 | def 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 | |
| 37 | def accuracy(model, blob_in, blob_out, **kwargs): |
searching dependent graphs…