cyiheng

Thank you @Jonas1312 But I still got the error but it changed a little bit ... ``` Traceback (most recent call last): File "main.py", line 182, in main(args) File "main.py",...

Thank you again for your help ! I printed the size before as you indicated: ``` with torch.no_grad(): if z_trg is not None: s_trg = nets.mapping_network(z_trg, y_trg) else: # x_ref...