Spaces:
Running
Running
File size: 728 Bytes
5e014de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
from easydict import EasyDict as edict
import numpy as np
__C = edict()
cfg = __C
# 0. basic config
__C.TAG = 'default'
__C.CLASSES = ['Real', 'Fake']
# config of network input
__C.MULTIMODAL_FUSION = edict()
__C.MULTIMODAL_FUSION.IMG_CHANNELS = [3, 64, 128, 256, 512]
__C.MULTIMODAL_FUSION.DCT_CHANNELS = [1, 64, 128, 256, 512]
__C.NUM_EPOCHS = 100
__C.BATCH_SIZE = 64
__C.NUM_WORKERS = 4
__C.LEARNING_RATE = 0.0001
__C.PRETRAINED = False
__C.PRETRAINED_PATH = "/home/user/Documents/Real_and_DeepFake/src/best_model.pth"
__C.TEST_BATCH_SIZE = 512
__C.TEST_CSV = "/home/user/Documents/Real_and_DeepFake/src/dataset/extended_val.csv"
__C.MODEL_PATH = "/home/user/Documents/Real_and_DeepFake/src/best_model.pth"
|