Commit
·
d5a0b8d
1
Parent(s):
5cb0801
updated generate 4ch huggingface
Browse files- generate_4ch_from_huggingface.py +1 -1
- test_out/0_img.png +0 -0
- test_out/0_mask.png +0 -0
generate_4ch_from_huggingface.py
CHANGED
|
@@ -52,7 +52,7 @@ if __name__ == "__main__":
|
|
| 52 |
parser = argparse.ArgumentParser(
|
| 53 |
description='generate images'
|
| 54 |
)
|
| 55 |
-
parser.add_argument('--ckpt', type=str, default="/work/vajira/DL/FastGAN-pytorch/train_results/test1_4ch/models/all_50000.pth")
|
| 56 |
parser.add_argument('--artifacts', type=str, default=".", help='path to artifacts.')
|
| 57 |
parser.add_argument('--cuda', type=int, default=0, help='index of gpu to use')
|
| 58 |
parser.add_argument('--start_iter', type=int, default=6)
|
|
|
|
| 52 |
parser = argparse.ArgumentParser(
|
| 53 |
description='generate images'
|
| 54 |
)
|
| 55 |
+
#parser.add_argument('--ckpt', type=str, default="/work/vajira/DL/FastGAN-pytorch/train_results/test1_4ch/models/all_50000.pth")
|
| 56 |
parser.add_argument('--artifacts', type=str, default=".", help='path to artifacts.')
|
| 57 |
parser.add_argument('--cuda', type=int, default=0, help='index of gpu to use')
|
| 58 |
parser.add_argument('--start_iter', type=int, default=6)
|
test_out/0_img.png
CHANGED
|
|
test_out/0_mask.png
CHANGED
|
|