-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathargs.py
33 lines (29 loc) · 2.06 KB
/
args.py
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
import argparse
def parameter_parser():
parser = argparse.ArgumentParser()
parser.add_argument('--cuda', action='store_true', default=True,
help='enables CUDA training')
parser.add_argument("--path", type=str, default="F:\wxh_work\datasets\MultiView_Dataset", help="Dataset path")
parser.add_argument("--dataset", type=str, default="yale", help="Dataset name")
parser.add_argument("--seed", type=int, default=42, help="Random seed")
parser.add_argument("--fix_seed", action='store_true', default=True, help="xx")
parser.add_argument("--n_repeated", type=int, default=10, help="Repeated times")
parser.add_argument("--lr", type=float, default=1e-3, help="Learning rate")
parser.add_argument("--weight_decay", type=float, default=5e-5, help="Weight decay")
parser.add_argument("--num_epoch", type=int, default=250, help="Training epochs")
parser.add_argument("--lambda1", type=float, default=2, help="Value of Lambda")
parser.add_argument("--lambda2", type=float, default=100, help="Value of Lambda")
parser.add_argument("--d1", type=int, default=128, help="hidden dimensions")
parser.add_argument("--d2", type=int, default=64, help="hidden dimensions")
parser.add_argument("--d3", type=int, default=32, help="hidden dimensions")
parser.add_argument("--d4", type=int, default=16, help="hidden dimensions")
parser.add_argument("--d5", type=int, default=32, help="hidden dimensions")
parser.add_argument("--d6", type=int, default=16, help="hidden dimensions")
parser.add_argument('--label_ratio', type=float, default=0.05,
help='Ratio of Labeled Samples for Validation (default: 0.10)')
parser.add_argument('--top_ratio', type=float, default=0.01,
help='Ratio of Labeled Samples for Validation (default: 0.10)')
parser.add_argument('--select_each_ratio', type=float, default=0.025,
help='Ratio of Labeled Samples for Validation (default: 0.10)')
args = parser.parse_args()
return args