Send feedback
  
   
 
  
    
      mediapipe_model_maker.face_stylizer.HParams
     
  
      
     
  
  
  
   
  
    
  
  
    
    
The hyperparameters for training face stylizers.
Inherits From: BaseHParams
mediapipe_model_maker . face_stylizer . HParams ( 
    learning_rate :  float  =  0.0008 , 
    batch_size :  int  =  4 , 
    epochs :  int  =  100 , 
    steps_per_epoch :  Optional [ int ]  =  None , 
    class_weights :  Optional [ Mapping [ int ,  float ]]  =  None , 
    shuffle :  bool  =  False , 
    repeat :  bool  =  False , 
    export_dir :  str  =  tempfile . mkdtemp (), 
    distribution_strategy :  str  =  'off' , 
    num_gpus :  int  =  0 , 
    tpu :  str  =  '' 
) 
Attributes 
learning_rate 
Learning rate to use for gradient descent training.
 
 
batch_size 
Batch size for training.
 
 
epochs 
Number of training iterations.
 
 
beta_1 
beta_1 used in tf.keras.optimizers.Adam.
 
 
beta_2 
beta_2 used in tf.keras.optimizers.Adam.
 
 
steps_per_epoch 
Dataclass field
 
 
class_weights 
Dataclass field
 
 
shuffle 
Dataclass field
 
 
repeat 
Dataclass field
 
 
export_dir 
Dataclass field
 
 
distribution_strategy 
Dataclass field
 
 
num_gpus 
Dataclass field
 
 
tpu 
Dataclass field
 
 
Methods 
get_strategyView source 
get_strategy () 
__eq__
__eq__ ( 
    other 
) 
Class Variables 
batch_size 
4
 
 
beta_1 
0.0
 
 
beta_2 
0.99
 
 
class_weights 
None
 
 
distribution_strategy 
'off'
 
 
epochs 
100
 
 
export_dir 
'/tmpfs/tmp/tmpnt_h4p9w'
 
 
learning_rate 
0.0008
 
 
num_gpus 
0
 
 
repeat 
False
 
 
shuffle 
False
 
 
steps_per_epoch 
None
 
 
tpu 
''
 
 
  
  
 
  
    
    
      
    
     
  
       
         
  
  
    
    Send feedback
  
   
 
       
    
    
  
  
 
  Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License , and code samples are licensed under the Apache 2.0 License . For details, see the Google Developers Site Policies . Java is a registered trademark of Oracle and/or its affiliates.
  Last updated 2024-05-07 UTC.
 
 
  
  
    
    
    
      
  
  
    Need to tell us more?
  
   
 
     
  
  
    
      [[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-05-07 UTC."],[],[]]