Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
luoyetx committed May 9, 2018
1 parent 8ef2656 commit f4f843b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion jfda/minibatch.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

class MiniBatcher(multiprocessing.Process):
'''generate minibatch
given a queue, put (positives, negatives, part faces, landmark faces) = (n1, n2, n3, n4)
given a queue, put (negatives, positives, part faces, landmark faces) = (n1, n2, n3, n4)
'''

def __init__(self, db_names, ns, net_type):
Expand Down
2 changes: 1 addition & 1 deletion layers/data_layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ def set_batch_num(self, n1, n2, n3, n4):
self.net_input_size = cfg.NET_INPUT_SIZE[cfg.NET_TYPE]

def set_data_queue(self, queue):
'''the queue should put a minibatch with size of (positives, negatives, part faces, landmark faces) =
'''the queue should put a minibatch with size of (negatives, positives, part faces, landmark faces) =
(n1, n2, n3, n4) in a dict
'''
self.data_queue = queue
Expand Down

0 comments on commit f4f843b

Please sign in to comment.