From 78d5150d83e74432d97f06ac1030e6c272f5a78b Mon Sep 17 00:00:00 2001
From: Raul Puri <raulpuric@berkeley.edu>
Date: Sat, 5 May 2018 17:23:11 -0700
Subject: [PATCH] inference (distributed) dataparallel patch

removing the '.module' that comes from (distibuted)dataparallel state dict
---
 inference.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference.ipynb b/inference.ipynb
index 26e38d1..7785cd2 100644
--- a/inference.ipynb
+++ b/inference.ipynb
@@ -98,7 +98,7 @@
    "source": [
     "checkpoint_path = \"/home/scratch.adlr-gcf/audio_denoising/runs/TTS-Tacotron2-LJS-MSE-DRC-NoMaskPadding-Unsorted-Distributed-22khz/checkpoint_15500\"\n",
     "model = load_model(hparams)\n",
-    "model.load_state_dict(torch.load(checkpoint_path)['state_dict'])\n",
+    "model.load_state_dict({k.replace('module.',''):v for k,v in torch.load(checkpoint_path)['state_dict'].items()})\n",
     "model = model.module\n",
     "_ = model.eval()"
    ]