Fix deprecated Timm import: timm.models.layers -> timm.layers

Author: deruyter92

animals/models/model_animals.py

@@ -13,7 +13,7 @@
 import torch
 import torch.nn as nn
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 from fmpose3d.animals.models.graph_frames import Graph
 

fmpose3d/animals/models/model_animal3d.py

@@ -13,7 +13,7 @@
 import torch
 import torch.nn as nn
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 from fmpose3d.animals.models.graph_frames import Graph
 from fmpose3d.models.base_model import BaseModel, register_model

fmpose3d/models/fmpose3d/model_GAMLP.py

@@ -15,7 +15,7 @@
 from fmpose3d.models.base_model import BaseModel, register_model
 from functools import partial
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 class TimeEmbedding(nn.Module):
     def __init__(self, dim: int, hidden_dim: int = 64):