NimaBoscarino commited on
Commit
5bc2e95
1 Parent(s): 9a4fd71
Files changed (3) hide show
  1. app.py +3 -1
  2. climategan/generator.py +1 -0
  3. climategan/trainer.py +1 -0
app.py CHANGED
@@ -1,5 +1,7 @@
1
  import os
2
  os.system('gdown https://drive.google.com/u/0/uc?id=18OCUIy7JQ2Ow_-cC5xn_hhDn-Bp45N1K')
3
  os.system('unzip release-github-v1.zip')
4
- os.system('python apply_events.py -b 1 -i ./inputs -r model/masker --output_path ./outputs --overwrite')
 
 
5
 
1
  import os
2
  os.system('gdown https://drive.google.com/u/0/uc?id=18OCUIy7JQ2Ow_-cC5xn_hhDn-Bp45N1K')
3
  os.system('unzip release-github-v1.zip')
4
+ os.system('mkdir config')
5
+ os.system('mv model config')
6
+ os.system('python apply_events.py -b 1 -i ./inputs -r config/model/masker --output_path ./outputs --overwrite')
7
 
climategan/generator.py CHANGED
@@ -367,6 +367,7 @@ class OmniGenerator(nn.Module):
367
 
368
  # path exists
369
  ckpt_path = Path(self.opts.val.val_painter).resolve()
 
370
  assert ckpt_path.exists()
371
 
372
  # path is a checkpoint path
367
 
368
  # path exists
369
  ckpt_path = Path(self.opts.val.val_painter).resolve()
370
+ print(ckpt_path)
371
  assert ckpt_path.exists()
372
 
373
  # path is a checkpoint path
climategan/trainer.py CHANGED
@@ -494,6 +494,7 @@ class Trainer:
494
 
495
  # specified m
496
  elif str(m_path) != "none":
 
497
  assert m_path.exists()
498
  assert "m" in self.opts.tasks
499
  model = "M"
494
 
495
  # specified m
496
  elif str(m_path) != "none":
497
+ print(m_path)
498
  assert m_path.exists()
499
  assert "m" in self.opts.tasks
500
  model = "M"