-
Notifications
You must be signed in to change notification settings - Fork 0
/
arguments.py
41 lines (35 loc) · 1.92 KB
/
arguments.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
34
35
36
37
38
39
40
41
import argparse
def Arguments():
parser = argparse.ArgumentParser()
parser.add_argument('--msg_k_dim', type=int, default=5,
help='key dimension')
parser.add_argument('--msg_v_dim', type=int, default=10,
help='value dimension')
parser.add_argument('--num_agents', type=int, default=5,
help='number of agents')
parser.add_argument('--encoded_size', type=int, default=20,
help='output dimension of encoder')
parser.add_argument('--prize', type=int, default=0,
help='reward on reaching consensus')
parser.add_argument('--penalize', type=int, default=-1,
help='penalize for not reaching consensus during the episode')
parser.add_argument('--threshold', type=int, default=2,
help='minimum value to reach consensus')
parser.add_argument('--temp', type=float, default=0.5,
help='tau(temperature) (default: 1.0)')
parser.add_argument('--text_hidden', type=int, default=20,
help='hidden dimension of character encoder')
parser.add_argument('--flag', type=int, default=1,
help='0: image encoder; 1: text encoder')
# ------------------------------------------------------------ #
parser.add_argument('--num_episodes', type=int, default=50000,
help='number of episodes')
parser.add_argument('--num_steps', type=int, default=25,
help='length of each episode')
parser.add_argument('--num_classes', type=int, default=10,
help='number of classes')
parser.add_argument('--learning_rate', type=float, default=1e-4,
help='learning rate')
parser.add_argument('--gamma', type=float, default=0.9,
help='discount factor')
return parser.parse_args()