<filename>models.py
import torch
import torch.nn as nn
import torch.nn.functional as F


class MyModel(nn.Module):
    def __init__(self,):
        raise NotImplementedError
    def forward(self,):
        raise NotImplementedError

<filename>models.py
import torch
import torch.nn as nn
import torch.nn.functional as F


class MyModel(nn.Module):
    def __init__(self,):
        raise NotImplementedError
    def forward(self,):
        raise NotImplementedError
