Spaces:

hugging-apps
/

exmr-motion-refinement

Running on Zero

exmr-motion-refinement / lib /sapiens /engine /optim /optimizer.py

Upload folder using huggingface_hub

762d5e5 verified 8 days ago

419 Bytes

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from sapiens.registry import OPTIMIZERS
from torch.optim import Adam, AdamW, SGD

OPTIMIZERS.register_module(name="AdamW")(AdamW)
OPTIMIZERS.register_module(name="Adam")(Adam)
OPTIMIZERS.register_module(name="SGD")(SGD)

	# Copyright (c) Meta Platforms, Inc. and affiliates.
	# All rights reserved.
	#
	# This source code is licensed under the license found in the
	# LICENSE file in the root directory of this source tree.

	from sapiens.registry import OPTIMIZERS
	from torch.optim import Adam, AdamW, SGD

	OPTIMIZERS.register_module(name="AdamW")(AdamW)
	OPTIMIZERS.register_module(name="Adam")(Adam)
	OPTIMIZERS.register_module(name="SGD")(SGD)