github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-10 23:33:30 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix Sorter.py

Browse source
This commit is contained in:
Colombo 2020-02-23 20:19:13 +04:00
parent ed384f52f0
commit 4ea70864fc
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
mainscripts/Sorter.py
View file

@ -723,11 +723,11 @@ def sort_by_absdiff(input_path):
image_paths = pathex.get_image_paths(input_path) image_paths = pathex.get_image_paths(input_path)
image_paths_len = len(image_paths) image_paths_len = len(image_paths)
batch_size = 1024 batch_size = 512
batch_size_remain = image_paths_len % batch_size batch_size_remain = image_paths_len % batch_size
i_t = tf.placeholder (tf.float32, (None,256,256,3) ) i_t = tf.placeholder (tf.float32, (None,None,None,None) )
j_t = tf.placeholder (tf.float32, (None,256,256,3) ) j_t = tf.placeholder (tf.float32, (None,None,None,None) )
outputs_full = [] outputs_full = []
outputs_remain = [] outputs_remain = []