dookiethedog

I also tried in the train.py file to change the line `@click.option('--data', help='Training data', metavar='[ZIP|DIR]', type=click.Path(exists=True, dir_okay=False), required=True) ` back to stylegan3's original `type=click.Path(exists=True, dir_okay=False), required=True)` but that gives me...

I only wanted to use this fork as it supports png's with transparent backgrounds. I just decided to dig into your code and change my augment.py, training_loop, and dataset_tool.py in...

> Sorry, I hadn't seen this issue before. I'd like to fix the issue, but since the RGBA data I tested with worked fine, I think you can give me...

Alright I figured out the solution. Changes to agent.py Add the below right after line "self.model..." `self.model.to('cuda:0')` Modify def get_action, in the else (add .to('cuda:0') `state0 = torch.tensor(state, dtype=torch.float).to('cuda:0')` Changes...