Hugging Face's logo

hikmatfarhat
/
wgan-gp

Feature Extraction
Transformers
Safetensors
WGAN_GP
custom_code
Model card Files
xet
 Community
wgan-gp
File size: 1,648 Bytes
cd4cfe7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
{
  "architectures": [
    "WGAN_GP"
  ],
  "auto_map": {
    "AutoConfig": "config.Config",
    "AutoModel": "model.WGAN_GP"
  },
  "batch_size": 256,
  "cfg": {
    "batch_size": 256,
    "comet_project": "wgan-gp",
    "comet_workspace": "wgan",
    "d_iter_per_g": 5,
    "data_dir": "/home/user/PyTorch-Lightning-GAN/GAN/celeba",
    "device": "cuda",
    "epochs": 500,
    "final_activation": {
      "g": "tanh"
    },
    "images_dir": "images",
    "img_ch": 3,
    "imsize": 64,
    "lr": {
      "d": 0.0001,
      "g": 0.0001
    },
    "name": "WGAN-GP",
    "norm_type": {
      "d": "GroupNorm",
      "g": "GroupNorm"
    },
    "num_sample_epochs": 20,
    "num_workers": 4,
    "resume": false,
    "samples_dir": "samples",
    "save_image_freq": 50,
    "save_model_freq": 100,
    "seed": 42,
    "use_fabric": false,
    "w_gp": 10,
    "weights_dir": "chkpt",
    "zdim": 128
  },
  "comet_project": "wgan-gp",
  "comet_workspace": "wgan",
  "d_iter_per_g": 5,
  "data_dir": "/home/user/PyTorch-Lightning-GAN/GAN/celeba",
  "device": "cuda",
  "epochs": 500,
  "final_activation": {
    "g": "tanh"
  },
  "images_dir": "images",
  "img_ch": 3,
  "imsize": 64,
  "lr": {
    "d": 0.0001,
    "g": 0.0001
  },
  "model_type": "WGAN_GP",
  "name": "WGAN-GP",
  "norm_type": {
    "d": "GroupNorm",
    "g": "GroupNorm"
  },
  "num_sample_epochs": 20,
  "num_workers": 4,
  "resume": false,
  "samples_dir": "samples",
  "save_image_freq": 50,
  "save_model_freq": 100,
  "seed": 42,
  "torch_dtype": "float32",
  "transformers_version": "4.35.2",
  "use_fabric": false,
  "w_gp": 10,
  "weights_dir": "chkpt",
  "zdim": 128
}