options.py

import argparse

parser = argparse.ArgumentParser(description='WTALC')
parser.add_argument('--path-dataset', type=str, default='/home/share/fating/ProcessDataset/Thumos14', help='the path of data feature')
parser.add_argument('--lr', type=float, default=0.00003,help='learning rate (default: 0.0001)')
parser.add_argument('--batch-size', type=int, default=10, help='number of instances in a batch of data (default: 10)')
parser.add_argument('--model-name', default='weakloc', help='name to save model')
parser.add_argument('--pretrained-ckpt', default=None, help='ckpt for pretrained model')
parser.add_argument('--feature-size', default=2048, help='size of feature (default: 2048)')
parser.add_argument('--num-class', type=int,default=20, help='number of classes (default: )')
parser.add_argument('--dataset-name', default='Thumos14reduced', help='dataset to train on (default: )')
parser.add_argument('--max-seqlen', type=int, default=560, help='maximum sequence length during training (default: 750)')
parser.add_argument('--num-similar', default=3, type=int,help='number of similar pairs in a batch of data  (default: 3)')
parser.add_argument('--similar-size',default=2,type=int,help='number of similar size, default: 2')
parser.add_argument('--seed', type=int, default=3552, help='random seed (default: 1)')
parser.add_argument('--max-iter', type=int, default=10000, help='maximum iteration to train (default: 50000)')
parser.add_argument('--feature-type', type=str, default='I3D', help='type of feature to be used I3D or UNT (default: I3D)')
parser.add_argument('--use-model',type=str,help='model used to train the network')
parser.add_argument('--interval', type=int, default=200,help='time interval of performing the test')
parser.add_argument('--momentum', type=float, default=.9)
parser.add_argument('--weight_decay', type=float, default=1e-3)
parser.add_argument('--dataset',type=str,default='SampleDataset')

#for proposal genration
parser.add_argument('--scale',type=float,default=1)
parser.add_argument("--feature_fps", type=int, default=25)
parser.add_argument('--gamma-oic', type=float, default=0.2)

parser.add_argument('--tIoU_thresh_params',type=float,nargs=3,default=[0.1,0.9,10])
parser.add_argument('--k',type=float,default=7)

parser.add_argument("--lambda1", type=float, default=1)
parser.add_argument("--lambda2", type=float, default=1)
parser.add_argument('--lambda3',type=float,default=0.01)
parser.add_argument('--lambda4',type=float,default=1)
parser.add_argument('--alpha',type=float,default=0)

parser.add_argument('--dropout_ratio',type=float,default=0.7)

parser.add_argument('--gamma',type=float,default=0.5)

# for pooling kernel size calculate
parser.add_argument('--t',type=int,default=4)

parser.add_argument('--lambd',type=float,default=1)
parser.add_argument('--sample_mode',type=str,default='All')
#----------------hamnet--------------------
parser.add_argument('--factor',type=float,default=0.5)