File size: 264 Bytes
402b016
d30298c
 
 
402b016
 
d30298c
 
402b016
d30298c
1
2
3
4
5
6
7
8
9
10
11
from transformers import PretrainedConfig
import torch


class DeepFakeConfig(PretrainedConfig):
    model_type = "ResNet"

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'