vlbthambawita commited on
Commit
d5a0b8d
1 Parent(s): 5cb0801

updated generate 4ch huggingface

Browse files
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