vlbthambawita
commited on
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