diff --git a/.compute b/.compute
index c1957615..4df52e0f 100644
--- a/.compute
+++ b/.compute
@@ -4,4 +4,4 @@ pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m
 yes | apt-get install espeak 
 python3 setup.py develop
 # python3 distribute.py --config_path config_cluster.json  --data_path ${SHARED_DIR}/data/keithito/LJSpeech-1.1/  --restore_path ${USER_DIR}/best_model.pth.tar
-python3 train.py --config_path config.json  --data_path ${SHARED_DIR}/data/keithito/LJSpeech-1.1/
\ No newline at end of file
+python3 train.py --config_path config.json  --data_path ${SHARED_DIR}/data/keithito/LJSpeech-1.1/ --output_path ../keep/  --restore_path ${USER_DIR}/best_model_by_tilman.pth.tar
\ No newline at end of file
diff --git a/train.py b/train.py
index 783b9c89..2b4226b8 100644
--- a/train.py
+++ b/train.py
@@ -407,7 +407,7 @@ def main(args):
         checkpoint = torch.load(args.restore_path)
         try:
             model.load_state_dict(checkpoint['model'])
-            optimizer.load_state_dict(checkpoint['optimizer'])
+            # optimizer.load_state_dict(checkpoint['optimizer'])
         except:
             print(" > Partial model initialization.")
             partial_init_flag = True