7c1c90f 2374ae7
1
2
3
4
5
6
7
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'
