CoCalc -- LayerBase.py

GitHub Repository: iperov/deepfacelab
Path: blob/master/core/leras/layers/LayerBase.py
⁶²⁸ views

1
from core.leras import nn
2
tf = nn.tf
3

4
class LayerBase(nn.Saveable):
5
    #override
6
    def build_weights(self):
7
        pass
8
    
9
    #override
10
    def forward(self, *args, **kwargs):
11
        pass
12
    
13
    def __call__(self, *args, **kwargs):
14
        return self.forward(*args, **kwargs)
15

16
nn.LayerBase = LayerBase
17