Python get_op_from_mode Examples

Example #1

File: network.py Project: AriChow/fastestimator

 def _get_signature_epoch(self, mode):
     signature_epoch = [0]
     mode_ops = get_op_from_mode(self.ops, mode)
     for op in mode_ops:
         if isinstance(op, Scheduler):
             signature_epoch.extend(op.keys)
     return list(set(signature_epoch)), mode_ops

Example #2

 def _get_signature_epoch(self, mode):
     signature_epoch = []
     if isinstance(self.batch_size, Scheduler):
         signature_epoch.extend(self.batch_size.keys)
     mode_ops = get_op_from_mode(self.ops, mode)
     for op in mode_ops:
         if isinstance(op, Scheduler):
             signature_epoch.extend(op.keys)
     if len(signature_epoch) == 0:
         signature_epoch.append(0)
     return list(set(signature_epoch)), mode_ops

Example #3

 def _check_ops(self, mode):
     if self.ops_local:
         self.mode_ops[mode] = get_op_from_mode(self.ops_local, mode)
         if len(self.mode_ops[mode]) > 0:
             verify_ops(self.mode_ops[mode], "RecordWriter")