"docs/source/vscode:/vscode.git/clone" did not exist on "0d1cff119578b04f52249670f0fc58abadfc832d"
  • Sayak Paul's avatar
    Add TFData2VecVision for semantic segmentation (#17271) · 9d99489f
    Sayak Paul authored
    
    
    * feat: initial implementation of data2vec segmentation model in TF.
    
    * chore: minor corrections to make the segmenter work.
    
    * chore: removed unncessary files.
    
    * chore: add tests and other modifications.
    
    * fix: loss computation for segmentation.
    
    * chore: remove unused variable.
    
    * chore: formatting.
    
    * added a dummy adaptive pooling layer.
    
    * removed unnecessary file.
    
    * potentially add identifiers to layer names.
    
    * fix: layer naming.
    
    * chore: removed unnecessary print.
    
    * Skipping unneeded test
    
    * chore: add logging to debug tolerance.
    
    * fix: segmentation tests for tfdata2vecvision
    
    * chore: make style.
    
    * fix: layer names, assertion to be resolved.
    
    * Bumping test tolerance a bit
    
    * chore: bump the tol in PT test.
    Co-authored-by: default avatarmatt <rocketknight1@gmail.com>
    9d99489f
data2vec.mdx 5.15 KB