https://github.com/ChrisWu1997/2D-Motion-Retargeting
Tip revision: 558187d552549d1e601a7ceef62b1c532b941c85 authored by dependabot[bot] on 21 June 2022, 21:57:02 UTC
Bump numpy from 1.16.2 to 1.22.0
Bump numpy from 1.16.2 to 1.22.0
Tip revision: 558187d
__init__.py
from agent.agents import Agent2x, Agent3x
def get_training_agent(config, net):
assert config.name is not None
if config.name == 'skeleton' or config.name == 'view':
return Agent2x(config, net)
else:
return Agent3x(config, net)
