File size: 195 Bytes
e34b94f
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
import torch.nn as nn
from abc import ABC

class BaseModel(nn.Module, ABC):
    def __init__(self):
        super().__init__()
    
    @classmethod
    def from_config(cls, config):
        pass