Upload DeepFakeECGFromPulse2Pulse

modeling_deepfake.py

@@ -285,8 +285,17 @@ class DeepFakeECGFromPulse2Pulse(PreTrainedModel):
                  latent_dim=100, post_proc_filt_len=512,
                  verbose=False, upsample=True)
         
-    def forward(self, tensor, labels=None):
-        x = self.model(tensor)
+    def forward(self, num_samples, labels=None):
         
-        return {"x": x}
+        outputs = []
+        
+        for i in range(num_samples):
+            noise = torch.Tensor(1, 8, 5000).uniform_(-1, 1)
+            x = self.model(noise)
+            x = x*6000
+            x = x.int()
+            x = torch.t(x.squeeze())
+            outputs.append(x)
+        
+        return outputs