Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion animals/models/model_animals.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion fmpose3d/animals/models/model_animal3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion fmpose3d/lib/hrnet/lib/models/pose_hrnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -493,7 +493,7 @@ def init_weights(self, pretrained=''):
need_init_state_dict = {}
for name, m in pretrained_state_dict.items():
if name.split('.')[0] in self.pretrained_layers \
or self.pretrained_layers[0] is '*':
or self.pretrained_layers[0] == '*':
need_init_state_dict[name] = m
self.load_state_dict(need_init_state_dict, strict=False)
elif pretrained:
Expand Down
2 changes: 1 addition & 1 deletion fmpose3d/models/fmpose3d/model_GAMLP.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
Loading