<filename>test_config.py<gh_stars>1-10
import os
from easydict import EasyDict as edict
import time
import torch

# init
__C = edict()
cfg = __C

# ------------------------------TEST------------------------
# if train batch=1, use only one GPU!!
__C.GPU_ID = [0] 		    # sigle gpu: [0], [1] ...; multi gpus: [0,1]

__C.NET = 'VGG16_LCM_REG' 	# net selection as train_config.py

__C.DATASET = 'QNRF' 	    # SHHA, SHHB, QNRF, UCF50

                            # testing model path
__C.MODEL_PATH = './exp/03-11_17-51_SHHA_VGG16_DM_1e-05/all_ep_218_mae_71.63_mse_116.86.pth'

__C.LOG_PARA = 100.

if __C.DATASET == 'UCF50':  # only for UCF50
    from datasets.UCF50.setting import cfg_data
    __C.VAL_INDEX = cfg_data.VAL_INDEX

<filename>test_config.py<gh_stars>1-10
import os
from easydict import EasyDict as edict
import time
import torch

# init
__C = edict()
cfg = __C

# ------------------------------TEST------------------------
# if train batch=1, use only one GPU!!
__C.GPU_ID = [0] 		    # sigle gpu: [0], [1] ...; multi gpus: [0,1]

__C.NET = 'VGG16_LCM_REG' 	# net selection as train_config.py

__C.DATASET = 'QNRF' 	    # SHHA, SHHB, QNRF, UCF50

                            # testing model path
__C.MODEL_PATH = './exp/03-11_17-51_SHHA_VGG16_DM_1e-05/all_ep_218_mae_71.63_mse_116.86.pth'

__C.LOG_PARA = 100.

if __C.DATASET == 'UCF50':  # only for UCF50
    from datasets.UCF50.setting import cfg_data
    __C.VAL_INDEX = cfg_data.VAL_INDEX

# init # ------------------------------TEST------------------------ # if train batch=1, use only one GPU!! # sigle gpu: [0], [1] ...; multi gpus: [0,1] # net selection as train_config.py # SHHA, SHHB, QNRF, UCF50 # testing model path # only for UCF50