Skip to content

Optimizer

Methods and Attributes

Bases: Module

Source code in torchebm/core/base_optimizer.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class Optimizer(nn.Module):
    def __init__(self):
        super().__init__()

    @abstractmethod
    def step(self):
        pass

    @abstractmethod
    def zero_grad(self):
        pass

    @abstractmethod
    def state_dict(self):
        pass

    @abstractmethod
    def load_state_dict(self, state_dict):
        pass

    @abstractmethod
    def to(self, device):
        pass

step abstractmethod 

step()
Source code in torchebm/core/base_optimizer.py 
12
13
14
@abstractmethod
def step(self):
    pass

zero_grad abstractmethod 

zero_grad()
Source code in torchebm/core/base_optimizer.py 
16
17
18
@abstractmethod
def zero_grad(self):
    pass

state_dict abstractmethod 

state_dict()
Source code in torchebm/core/base_optimizer.py 
20
21
22
@abstractmethod
def state_dict(self):
    pass

load_state_dict abstractmethod 

load_state_dict(state_dict)
Source code in torchebm/core/base_optimizer.py 
24
25
26
@abstractmethod
def load_state_dict(self, state_dict):
    pass

to abstractmethod 

to(device)
Source code in torchebm/core/base_optimizer.py 
28
29
30
@abstractmethod
def to(self, device):
    pass