Code

Add Vision Transformer from ScaleMAE#422

Code

Draft

anwai98 wants to merge 5 commits intomainconstantinpape/torch-em:mainfrom vit-scale-maeconstantinpape/torch-em:vit-scale-maeCopy head branch name to clipboard

+250-12

Commits

Commits on Nov 29, 2024

• Add vision transformer of scale-mae

  anwai98

  committed
  7b60d6d

  Copy full SHA for 7b60d6d

• Restore vit_b default params

  anwai98

  committed
  14a0723

  Copy full SHA for 14a0723

• Add support for using skip connections

  anwai98

  committed
  dccdc7e

  Copy full SHA for dccdc7e

Commits on Jan 17, 2025

• Merge branch 'main' into vit-scale-mae

  anwai98

  committed
  f66c6d0

  Copy full SHA for f66c6d0

Commits on Jan 20, 2025

• Fix scale to (1.0, 1.0) to avoid cropping images at all

  Show description for f69eb16
  anwai98

  authored
  f69eb16

  Copy full SHA for f69eb16