From 9c7debe75cc7fe405a12871dc09856afd13202f7 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sat, 2 May 2020 17:47:25 -0600
Subject: [PATCH] Add colab option

---
 .../train/train_ESRGAN_blacked_colab.yml      | 84 +++++++++++++++++++
 codes/temp/cleanup.sh                         |  4 +
 2 files changed, 88 insertions(+)
 create mode 100644 codes/options/train/train_ESRGAN_blacked_colab.yml
 create mode 100644 codes/temp/cleanup.sh

diff --git a/codes/options/train/train_ESRGAN_blacked_colab.yml b/codes/options/train/train_ESRGAN_blacked_colab.yml
new file mode 100644
index 00000000..722ae647
--- /dev/null
+++ b/codes/options/train/train_ESRGAN_blacked_colab.yml
@@ -0,0 +1,84 @@
+#### general settings
+name: blacked_fix_and_upconv
+use_tb_logger: true
+model: srgan
+distortion: sr
+scale: 4
+gpu_ids: [0]
+amp_opt_level: O1
+
+#### datasets
+datasets:
+  train:
+    name: vixcloseup
+    mode: LQGT
+    dataroot_GT: /content/hr
+    dataroot_LQ: /content/lr_corrupted
+    doCrop: false
+    use_shuffle: true
+    n_workers: 12  # per GPU
+    batch_size: 12
+    target_size: 256
+    color: RGB
+  val:
+    name: adrianna_val
+    mode: LQGT
+    dataroot_GT: /content/val/hhq
+    dataroot_LQ: /content/val/hr
+
+#### network structures
+network_G:
+  which_model_G: RRDBNet
+  in_nc: 3
+  out_nc: 3
+  nf: 48
+  nb: 23
+network_D:
+  which_model_D: discriminator_resnet
+  in_nc: 3
+  nf: 48
+
+#### path
+path:
+  pretrain_model_G: experiments/7000_G.pth
+  pretrain_model_D: experiments/7000_D.pth
+  strict_load: true
+  resume_state: ~
+
+#### training settings: learning rate scheme, loss
+train:
+  lr_G: !!float 6e-5
+  weight_decay_G: 0
+  beta1_G: 0.9
+  beta2_G: 0.99
+  lr_D: !!float 8e-5
+  weight_decay_D: 0
+  beta1_D: 0.9
+  beta2_D: 0.99
+  lr_scheme: MultiStepLR
+
+  niter: 400000
+  warmup_iter: -1  # no warm up
+  lr_steps: [20000, 40000, 60000, 80000]
+  lr_gamma: 0.5
+
+  pixel_criterion: l1
+  pixel_weight: !!float 1e-2
+  feature_criterion: l1
+  feature_weight: 1
+  feature_weight_decay: 1
+  feature_weight_decay_steps: 500
+  feature_weight_minimum: 1
+  gan_type: gan  # gan | ragan
+  gan_weight: !!float 1e-2
+
+  D_update_ratio: 1
+  D_init_iters: 0
+
+  manual_seed: 10
+  val_freq: !!float 5e2
+
+#### logger
+logger:
+  print_freq: 50
+  save_checkpoint_freq: !!float 5e2
diff --git a/codes/temp/cleanup.sh b/codes/temp/cleanup.sh
new file mode 100644
index 00000000..95f03933
--- /dev/null
+++ b/codes/temp/cleanup.sh
@@ -0,0 +1,4 @@
+rm gen/*
+rm hr/*
+rm lr/*
+rm pix/*
\ No newline at end of file