Update train.py
This commit is contained in:
parent
9a9dd8eb3d
commit
3519dd6ba0
2
train.py
2
train.py
|
@ -842,7 +842,7 @@ def update_old_args(t_args):
|
||||||
t_args.__dict__["shuffle_tags"] = False
|
t_args.__dict__["shuffle_tags"] = False
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
supported_resolutions = [448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 1088, 1152]
|
supported_resolutions = [384, 448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 1088, 1152]
|
||||||
argparser = argparse.ArgumentParser(description="EveryDream2 Training options")
|
argparser = argparse.ArgumentParser(description="EveryDream2 Training options")
|
||||||
argparser.add_argument("--config", type=str, required=False, default=None, help="JSON config file to load options from")
|
argparser.add_argument("--config", type=str, required=False, default=None, help="JSON config file to load options from")
|
||||||
args, _ = argparser.parse_known_args()
|
args, _ = argparser.parse_known_args()
|
||||||
|
|
Loading…
Reference in New Issue