cc-ai / climategan

Code and pre-trained model for the algorithm generating visualisations of 3 climate change related events: floods, wildfires and smog.
https://thisclimatedoesnotexist.com
GNU General Public License v3.0
72 stars 18 forks source link

Victor's Ablation Study exps #189

Closed vict0rsch closed 3 years ago

vict0rsch commented 3 years ago
Exp. Comet Link Slurm JOBID comet.tags gen.m.use_spade gen.s.depth_dada_fusion gen.s.depth_feat_fusion partition tasks train.pseudo.tasks output_path
0 0 747235 m False False False unkilla.. m [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (13)
1 1 747236 md False False False main md [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (14)
2 2 747237 ms False False False main ms [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (15)
3 3 747238 msd False False False long msd [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (16)
4 4 747239 msd spade True False False long msd [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (17)
5 5 747240 msd dada False True True long msd [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (18)
6 6 747241 msd spade dada True True True long msd [] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (19)
7 7 747242 msd pseudo False False False long msd ['m', 's', 'd'] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (20)
8 8 747243 msd spade pseudo True False False long msd ['m', 's', 'd'] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (21)
9 9 747244 msd dada pseudo False True True long msd ['m', 's', 'd'] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (22)
10 10 747245 msd spade dada pseudo True True True long msd ['m', 's', 'd'] /miniscratch/schmidtv/vicc/omnigan/runs/ablation (23)
vict0rsch commented 3 years ago

Exp file:

partition:
  sample: sequential
  from:
    - unkillable
    - main
    - main
    - long
    - long
    - long
    - long
    - long
    - long
    - long
    - long
cpus: 4
gres: gpu:rtx8000:1
mem: 24G
output: /miniscratch/schmidtv/vicc/omnigan/logs/ablation/slurm-%j.out
codeloc: $HOME/ccai/github/omnigan
conda: conda activate omnienv && conda deactivate && conda activate omnienv
modules: module load anaconda/3

"args.config": $HOME/ccai/github/omnigan/shared/trainer/defaults.yaml

n_search: -1

"args.comet_tags": [ablation]

"data.loaders.batch_size": 4
"data.loaders.num_workers": 4

"gen.m.spade.cond_nc": 15

"output_path": /miniscratch/schmidtv/vicc/omnigan/runs/ablation

"train.epochs": 150
"train.kitti.pretrain": False
"train.pseudo.epochs": 10
"train.save_n_epochs": 25

"comet.tags":
  sample: sequential
  from:
    - [m]
    - [md]
    - [ms]
    - [msd]
    - [msd_spade]
    - [msd, dada]
    - [msd_spade, dada]
    - [msd, pseudo]
    - [msd_spade, pseudo]
    - [msd, dada, pseudo]
    - [msd_spade, dada, pseudo]

tasks:
  sample: sequential
  from:
    - [m]
    - [m, d]
    - [m, s]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]

"train.pseudo.tasks":
  sample: sequential
  from:
    - []
    - []
    - []
    - []
    - []
    - []
    - []
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]
    - [m, s, d]

"gen.m.use_spade":
  sample: sequential
  from:
    - False
    - False
    - False
    - False
    - True
    - False
    - True
    - False
    - True
    - False
    - True
"gen.deeplabv3.backbone": resnet
"gen.s.depth_feat_fusion":
  sample: sequential
  from:
    - False
    - False
    - False
    - False
    - False
    - True
    - True
    - False
    - False
    - True
    - True
"gen.s.depth_dada_fusion":
  sample: sequential
  from:
    - False
    - False
    - False
    - False
    - False
    - True
    - True
    - False
    - False
    - True
    - True