mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 10:55:57 +08:00
31 lines
552 B
Python
31 lines
552 B
Python
"""
|
|
Optimizer class.
|
|
"""
|
|
|
|
from torch import optim
|
|
|
|
|
|
class Optimizer:
|
|
"""_summary_"""
|
|
|
|
def __init__(self, params: dict, lr: float):
|
|
"""
|
|
Args:
|
|
params (dict): _description_
|
|
lr (float): Learning rate.
|
|
"""
|
|
self.optimizer = optim.Adam(params)
|
|
self.lr = lr
|
|
|
|
def test(self):
|
|
"""_summary_
|
|
|
|
Raises:
|
|
NotImplementedError: _description_
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
def test2(self):
|
|
"""Summary"""
|
|
raise NotImplementedError
|