-
Notifications
You must be signed in to change notification settings - Fork 65
Expand file tree
/
Copy pathglobals.py
More file actions
36 lines (27 loc) · 741 Bytes
/
globals.py
File metadata and controls
36 lines (27 loc) · 741 Bytes
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
"""
constants for the data set.
ModelNet40 for example
"""
NUM_CLASSES = 40
NUM_VIEWS = 12
TRAIN_LOL = './data/view/train_lists.txt'
VAL_LOL = './data/view/val_lists.txt'
TEST_LOL = './data/view/test_lists.txt'
"""
constants for both training and testing
"""
BATCH_SIZE = 16
# this must be more than twice the BATCH_SIZE
INPUT_QUEUE_SIZE = 4 * BATCH_SIZE
"""
constants for training the model
"""
INIT_LEARNING_RATE = 0.0001
# sample how many shapes for validation
# this affects the validation time
VAL_SAMPLE_SIZE = 256
# do a validation every VAL_PERIOD iterations
VAL_PERIOD = 100
# save the progress to checkpoint file every SAVE_PERIOD iterations
# this takes tens of seconds. Don't set it smaller than 100.
SAVE_PERIOD = 1000