From e82f4552db84b56b5813ecc757c4073c27634999 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Fri, 18 Dec 2020 16:21:28 -0700
Subject: [PATCH] Update other docs with dumb config options

---
 codes/train.py                        | 2 +-
 codes/utils/options.py                | 2 --
 recipes/byol/train_div2k_byol.yml     | 1 -
 recipes/srflow/train_div2k_srflow.yml | 5 +----
 4 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/codes/train.py b/codes/train.py
index e8a634a0..ed925a88 100644
--- a/codes/train.py
+++ b/codes/train.py
@@ -293,7 +293,7 @@ class Trainer:
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_mi1_rrdb_bigboi_pretrain.yml')
+    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_faces_glean.yml')
     parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
     parser.add_argument('--local_rank', type=int, default=0)
     args = parser.parse_args()
diff --git a/codes/utils/options.py b/codes/utils/options.py
index 64240ad8..8af09757 100644
--- a/codes/utils/options.py
+++ b/codes/utils/options.py
@@ -15,8 +15,6 @@ def parse(opt_path, is_train=True):
         print('export CUDA_VISIBLE_DEVICES=' + gpu_list)
 
     opt['is_train'] = is_train
-    if opt['distortion'] == 'sr' or opt['distortion'] == 'downsample':
-        scale = opt['scale']
 
     # datasets
     if 'datasets' in opt.keys():
diff --git a/recipes/byol/train_div2k_byol.yml b/recipes/byol/train_div2k_byol.yml
index 407b3d39..bac72712 100644
--- a/recipes/byol/train_div2k_byol.yml
+++ b/recipes/byol/train_div2k_byol.yml
@@ -2,7 +2,6 @@
 name: train_div2k_byol
 use_tb_logger: true
 model: extensibletrainer
-distortion: sr
 scale: 1
 gpu_ids: [0]
 fp16: false
diff --git a/recipes/srflow/train_div2k_srflow.yml b/recipes/srflow/train_div2k_srflow.yml
index 05d55cad..3da10899 100644
--- a/recipes/srflow/train_div2k_srflow.yml
+++ b/recipes/srflow/train_div2k_srflow.yml
@@ -2,7 +2,6 @@
 name: train_div2k_srflow
 use_tb_logger: true
 model: extensibletrainer
-distortion: sr
 scale: 4
 gpu_ids: [0]
 fp16: false
@@ -17,12 +16,10 @@ datasets:
     name: div2k
     mode: single_image_extensible
     paths: /content/div2k   # <-- Put your path here.
-    target_size: 160        # <-- SRFlow trains better with factors of 160 for some reason.
+    target_size: 160
     force_multiple: 1
     scale: 4
-    eval: False
     num_corrupts_per_image: 0
-    strict: false
 
 networks:
   generator: