Files
automatic/modules/dml/optimizer/optimizer.py
T
Seunghoon Lee 09ae33cdf7 Implement torch.dml.
VERY UNSTABLE & NOT TESTED.
2023-04-26 12:21:44 +09:00

9 lines
191 B
Python

from abc import *
from typing import *
class Optimizer(metaclass=ABCMeta):
driver: Any = None
@abstractmethod
def memory_stats(self, index: int) -> Tuple[int, int]:
pass