mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
fix
This commit is contained in:
parent
91187ecb95
commit
e53be5e22d
1 changed files with 5 additions and 4 deletions
|
@ -358,10 +358,11 @@ class AMPModel(ModelBase):
|
||||||
gpu_dst_inter_src_code, gpu_dst_inter_dst_code = self.inter_src (gpu_dst_code), self.inter_dst (gpu_dst_code)
|
gpu_dst_inter_src_code, gpu_dst_inter_dst_code = self.inter_src (gpu_dst_code), self.inter_dst (gpu_dst_code)
|
||||||
|
|
||||||
inter_dims_bin = int(inter_dims*morph_factor)
|
inter_dims_bin = int(inter_dims*morph_factor)
|
||||||
inter_rnd_binomial = tf.stack([tf.concat([tf.tile(tf.constant([1], tf.float32), ( inter_dims_bin, )),
|
with tf.device(f'/CPU:0'):
|
||||||
tf.tile(tf.constant([0], tf.float32), ( inter_dims-inter_dims_bin, ))], 0 ) for _ in range(bs_per_gpu)], 0)
|
inter_rnd_binomial = tf.stack([tf.random.shuffle(tf.concat([tf.tile(tf.constant([1], tf.float32), ( inter_dims_bin, )),
|
||||||
|
tf.tile(tf.constant([0], tf.float32), ( inter_dims-inter_dims_bin, ))], 0 )) for _ in range(bs_per_gpu)], 0)
|
||||||
inter_rnd_binomial = tf.stop_gradient(inter_rnd_binomial[...,None,None])
|
|
||||||
|
inter_rnd_binomial = tf.stop_gradient(inter_rnd_binomial[...,None,None])
|
||||||
|
|
||||||
gpu_src_code = gpu_src_inter_src_code * inter_rnd_binomial + gpu_src_inter_dst_code * (1-inter_rnd_binomial)
|
gpu_src_code = gpu_src_inter_src_code * inter_rnd_binomial + gpu_src_inter_dst_code * (1-inter_rnd_binomial)
|
||||||
gpu_dst_code = gpu_dst_inter_dst_code
|
gpu_dst_code = gpu_dst_inter_dst_code
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue