Spaces:
Build error
Build error
File size: 2,076 Bytes
711b041 |
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import argparse
import os
import random
import warnings
warnings.filterwarnings('ignore')
import numpy as np
import torch
from run import Run
parser = argparse.ArgumentParser()
parser.add_argument('--model_name', default='SVFEND', help='SVFEND/FANVM/C3D/VGG/Bbox/Vggish/Bert/TextCNN/Comments/TikTec')
parser.add_argument('--mode_eval', default= 'nocv', help='nocv/cv/temporal')
parser.add_argument('--fold', type=int, default= 1, help='needed when model_eval=nocv')
parser.add_argument('--epoches', type=int, default=30)
parser.add_argument('--batch_size', type = int, default=128)
parser.add_argument('--num_workers', type=int, default=0)
parser.add_argument('--epoch_stop', type=int, default=5)
parser.add_argument('--seed', type=int, default=2022)
parser.add_argument('--gpu', type=int, required=True)
parser.add_argument('--lr', type=float, default=0.0001)
parser.add_argument('--lambd', type=float, default=0.1)
parser.add_argument('--dropout', type=float, default=0.1)
parser.add_argument('--weight_decay', type=float, default=5e-5)
parser.add_argument('--path_param', default= './checkpoints/')
parser.add_argument('--path_tensorboard', default= './tb/')
args = parser.parse_args()
# os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu)
seed = args.seed
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.deterministic = True
print (args)
config = {
'model_name': args.model_name,
'mode_eval':args.mode_eval,
'fold':args.fold,
'epoches': args.epoches,
'batch_size': args.batch_size,
'num_workers': args.num_workers,
'epoch_stop': args.epoch_stop,
'seed': args.seed,
'device': args.gpu,
'lr': args.lr,
'lambd': args.lambd,
'dropout': args.dropout,
'weight_decay': args.weight_decay,
'path_param': args.path_param,
'path_tensorboard': args.path_tensorboard,
}
if __name__ == '__main__':
Run(config = config
).main()
|