(self, net)
| 213 | self._add_ops(net, self.param_blobs, predictor_fc_fp_version) |
| 214 | |
| 215 | def add_train_ops(self, net): |
| 216 | # use the train_param_blobs to be consistent with the SamplingTrain unittest |
| 217 | self._add_ops(net, self.train_param_blobs, "fp32") |
| 218 | |
| 219 | def get_fp16_compatible_parameters(self): |
| 220 | if self.output_dim_vec is None or len(self.output_dim_vec) == 1: |