piddnad / agilegan Goto Github PK

is there any tips on how to generate training data?

Thanks for your sharing!!
I have two questions about the code you shared:

1. DualGenerator in your code is only used in inference?
2. If not, can you share your training codes?

hi, after we trained model on own dataset, how to test the checkpoints/best_model.pt?
I have replaced the low_path='pretrain/pixar.pt' to my own checkpoint, and get the corresponding error:

  File "test.py", line 97, in <module>
    T=Tester(input_opts.model)
  File "test.py", line 26, in __init__
    g_ema.load_state_dict(checkpoint["g_ema"], strict=False)
KeyError: 'g_ema'```