comment out annoying print statement
This commit is contained in:
@@ -583,7 +583,7 @@ class VAE:
|
|||||||
|
|
||||||
def broadcast_image_to(tensor, target_batch_size, batched_number):
|
def broadcast_image_to(tensor, target_batch_size, batched_number):
|
||||||
current_batch_size = tensor.shape[0]
|
current_batch_size = tensor.shape[0]
|
||||||
print(current_batch_size, target_batch_size)
|
#print(current_batch_size, target_batch_size)
|
||||||
if current_batch_size == 1:
|
if current_batch_size == 1:
|
||||||
return tensor
|
return tensor
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user