• BasiaFusinska's avatar
    Merged commit includes the following changes: (#8753) · df89d3e0
    BasiaFusinska authored
    318938278  by Andre Araujo:
    
        Loading pretrained ImageNet weights to initialize the ResNet backbone. Changed the defaults of the batch size and initial learning rate to increase convergence on the GLDv2 dataset. Made the evaluation batch size dynamic depending on the global batch size.
    
    --
    318911740  by Andre Araujo:
    
        Introduced additional shuffling of the TRAIN and VALIDATION datasets to ensure label variance across batches.
    
    --
    318908335  by Andre Araujo:
    
        Export model migration to TF2.
    
    --
    318489123  by Andre Araujo:
    
        Model exporting script for global feature trained with DELF codebase.
        Additionally, makes a small change to replace back_prop=False in the tf.while_loop call (see deprecation notice in https://www.tensorflow.org/api_docs/python/tf/while_loop
    
    ).
    
    --
    318401984  by Andre Araujo:
    
        Add attention visualization to DELF training script.
    
    --
    318168500  by Andre Araujo:
    
        Several small changes to DELF open-source training code:
        - Replace "make_dataset_iterator" call which was deprecated by a more recent suitable version.
        - Add image summary, allowing visualization of the augmented images during training
        - Normalize images before feeding them to the model
    
    --
    316888714  by Andre Araujo:
    
        - Removed unnecessary cast from feature_aggregation_extraction.py
        - Fixed clustering script
    
    --
    
    PiperOrigin-RevId: 318938278
    Co-authored-by: default avatarAndre Araujo <andrearaujo@google.com>
    df89d3e0
train.py 17.9 KB