Update networks/generator.py

networks/generator.py

@@ -103,7 +103,7 @@ class Generator(nn.Module):
 
 		return vid_target_recon
 
-	def edit_vid_batch(self, img_source, vid_target, d_l, v_l, chunk_size):
+	def edit_vid_batch(self, vid_target, d_l, v_l, chunk_size):
 
 		b,t,c,h,w = vid_target.size()
 		img_source = vid_target[:, 0, :, :, :]