Пример #1
0
        public void MergeFrom(pb::CodedInputStream input)
        {
            uint tag;

            while ((tag = input.ReadTag()) != 0)
            {
                switch (tag)
                {
                default:
                    _unknownFields = pb::UnknownFieldSet.MergeFieldFrom(_unknownFields, input);
                    break;

                case 8: {
                    BatchSize = input.ReadUInt32();
                    break;
                }

                case 18: {
                    dataAugmentationOptions_.AddEntriesFrom(input, _repeated_dataAugmentationOptions_codec);
                    break;
                }

                case 24: {
                    SyncReplicas = input.ReadBool();
                    break;
                }

                case 37: {
                    KeepCheckpointEveryNHours = input.ReadFloat();
                    break;
                }

                case 42: {
                    if (optimizer_ == null)
                    {
                        optimizer_ = new global::Tensorflow.Models.ObjectDetection.Protos.Optimizer();
                    }
                    input.ReadMessage(optimizer_);
                    break;
                }

                case 53: {
                    GradientClippingByNorm = input.ReadFloat();
                    break;
                }

                case 58: {
                    FineTuneCheckpoint = input.ReadString();
                    break;
                }

                case 64: {
                    FromDetectionCheckpoint = input.ReadBool();
                    break;
                }

                case 72: {
                    NumSteps = input.ReadUInt32();
                    break;
                }

                case 85: {
                    StartupDelaySteps = input.ReadFloat();
                    break;
                }

                case 93: {
                    BiasGradMultiplier = input.ReadFloat();
                    break;
                }

                case 98: {
                    freezeVariables_.AddEntriesFrom(input, _repeated_freezeVariables_codec);
                    break;
                }

                case 104: {
                    ReplicasToAggregate = input.ReadInt32();
                    break;
                }

                case 112: {
                    BatchQueueCapacity = input.ReadInt32();
                    break;
                }

                case 120: {
                    NumBatchQueueThreads = input.ReadInt32();
                    break;
                }

                case 128: {
                    PrefetchQueueCapacity = input.ReadInt32();
                    break;
                }

                case 136: {
                    MergeMultipleLabelBoxes = input.ReadBool();
                    break;
                }

                case 144: {
                    AddRegularizationLoss = input.ReadBool();
                    break;
                }

                case 152: {
                    LoadAllDetectionCheckpointVars = input.ReadBool();
                    break;
                }

                case 160: {
                    MaxNumberOfBoxes = input.ReadInt32();
                    break;
                }

                case 168: {
                    UnpadGroundtruthTensors = input.ReadBool();
                    break;
                }

                case 178: {
                    FineTuneCheckpointType = input.ReadString();
                    break;
                }

                case 184: {
                    RetainOriginalImages = input.ReadBool();
                    break;
                }

                case 192: {
                    UseMulticlassScores = input.ReadBool();
                    break;
                }

                case 202: {
                    updateTrainableVariables_.AddEntriesFrom(input, _repeated_updateTrainableVariables_codec);
                    break;
                }

                case 208: {
                    UseBfloat16 = input.ReadBool();
                    break;
                }

                case 216: {
                    SummarizeGradients = input.ReadBool();
                    break;
                }
                }
            }
        }
Пример #2
0
 public void MergeFrom(TrainConfig other)
 {
     if (other == null)
     {
         return;
     }
     if (other.BatchSize != 0)
     {
         BatchSize = other.BatchSize;
     }
     dataAugmentationOptions_.Add(other.dataAugmentationOptions_);
     if (other.SyncReplicas != false)
     {
         SyncReplicas = other.SyncReplicas;
     }
     if (other.KeepCheckpointEveryNHours != 0F)
     {
         KeepCheckpointEveryNHours = other.KeepCheckpointEveryNHours;
     }
     if (other.optimizer_ != null)
     {
         if (optimizer_ == null)
         {
             optimizer_ = new global::Tensorflow.Models.ObjectDetection.Protos.Optimizer();
         }
         Optimizer.MergeFrom(other.Optimizer);
     }
     if (other.GradientClippingByNorm != 0F)
     {
         GradientClippingByNorm = other.GradientClippingByNorm;
     }
     if (other.FineTuneCheckpoint.Length != 0)
     {
         FineTuneCheckpoint = other.FineTuneCheckpoint;
     }
     if (other.FineTuneCheckpointType.Length != 0)
     {
         FineTuneCheckpointType = other.FineTuneCheckpointType;
     }
     if (other.FromDetectionCheckpoint != false)
     {
         FromDetectionCheckpoint = other.FromDetectionCheckpoint;
     }
     if (other.LoadAllDetectionCheckpointVars != false)
     {
         LoadAllDetectionCheckpointVars = other.LoadAllDetectionCheckpointVars;
     }
     if (other.NumSteps != 0)
     {
         NumSteps = other.NumSteps;
     }
     if (other.StartupDelaySteps != 0F)
     {
         StartupDelaySteps = other.StartupDelaySteps;
     }
     if (other.BiasGradMultiplier != 0F)
     {
         BiasGradMultiplier = other.BiasGradMultiplier;
     }
     updateTrainableVariables_.Add(other.updateTrainableVariables_);
     freezeVariables_.Add(other.freezeVariables_);
     if (other.ReplicasToAggregate != 0)
     {
         ReplicasToAggregate = other.ReplicasToAggregate;
     }
     if (other.BatchQueueCapacity != 0)
     {
         BatchQueueCapacity = other.BatchQueueCapacity;
     }
     if (other.NumBatchQueueThreads != 0)
     {
         NumBatchQueueThreads = other.NumBatchQueueThreads;
     }
     if (other.PrefetchQueueCapacity != 0)
     {
         PrefetchQueueCapacity = other.PrefetchQueueCapacity;
     }
     if (other.MergeMultipleLabelBoxes != false)
     {
         MergeMultipleLabelBoxes = other.MergeMultipleLabelBoxes;
     }
     if (other.UseMulticlassScores != false)
     {
         UseMulticlassScores = other.UseMulticlassScores;
     }
     if (other.AddRegularizationLoss != false)
     {
         AddRegularizationLoss = other.AddRegularizationLoss;
     }
     if (other.MaxNumberOfBoxes != 0)
     {
         MaxNumberOfBoxes = other.MaxNumberOfBoxes;
     }
     if (other.UnpadGroundtruthTensors != false)
     {
         UnpadGroundtruthTensors = other.UnpadGroundtruthTensors;
     }
     if (other.RetainOriginalImages != false)
     {
         RetainOriginalImages = other.RetainOriginalImages;
     }
     if (other.UseBfloat16 != false)
     {
         UseBfloat16 = other.UseBfloat16;
     }
     if (other.SummarizeGradients != false)
     {
         SummarizeGradients = other.SummarizeGradients;
     }
     _unknownFields = pb::UnknownFieldSet.MergeFrom(_unknownFields, other._unknownFields);
 }