feat: extract opengait_studio monorepo module
Move demo implementation into opengait_studio, retire Sports2D runtime integration, and align packaging with root-level monorepo dependency management.
This commit is contained in:
@@ -7,7 +7,7 @@ import torch.nn.functional as F
|
||||
from ..base_model import BaseModel
|
||||
from ..modules import PackSequenceWrapper, HorizontalPoolingPyramid, SetBlockWrapper, ParallelBN1d, SeparateFCs
|
||||
|
||||
from utils import np2var, list2var, get_valid_args, ddp_all_gather
|
||||
from opengait.utils import np2var, list2var, get_valid_args, ddp_all_gather
|
||||
from data.transform import get_transform
|
||||
from einops import rearrange
|
||||
|
||||
@@ -143,7 +143,7 @@ class GaitSSB_Pretrain(BaseModel):
|
||||
|
||||
import torch.optim as optim
|
||||
import numpy as np
|
||||
from utils import get_valid_args, list2var
|
||||
from opengait.utils import get_valid_args, list2var
|
||||
|
||||
class no_grad(torch.no_grad):
|
||||
def __init__(self, enable=True):
|
||||
|
||||
Reference in New Issue
Block a user