DOFA: save parameters as instance attributes (#2346)

* add embed_dim as attr

* also add depth

* add other attributes

* move params

* order

torchgeo/models/dofa.py

@@ -65,6 +65,13 @@ def __init__(
             num_layers: Number of layers.
         """
         super().__init__()
+
+        self.input_dim = input_dim
+        self.output_dim = output_dim
+        self.embed_dim = embed_dim
+        self.num_heads = num_heads
+        self.num_layers = num_layers
+
         encoder_layer = nn.TransformerEncoderLayer(
             d_model=input_dim,
             nhead=num_heads,
@@ -267,8 +274,17 @@ def __init__(
         """
         super().__init__()
 
+        self.img_size = img_size
+        self.patch_size = patch_size
+        self.drop_rate = drop_rate
+        self.embed_dim = embed_dim
+        self.depth = depth
+        self.num_heads = num_heads
         self.dynamic_embed_dim = dynamic_embed_dim
+        self.num_classes = num_classes
         self.global_pool = global_pool
+        self.mlp_ratio = mlp_ratio
+
         if self.global_pool:
             norm_layer = norm_layer
             embed_dim = embed_dim