ndacremont/go-ai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: create output folder

Browse Source
This commit is contained in:
Martin Eyben 2025-05-09 10:59:11 +02:00
parent 614bb5acb2
commit 4e411c260d
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
go_ia/main.py
View File

@ -1,6 +1,7 @@
#!python #!python
import time import time
from os import path from os import path
import os
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import argparse import argparse
import torch import torch
@ -279,6 +280,8 @@ if __name__ == "__main__":
mymodel = GoModel().to(device) mymodel = GoModel().to(device)
output_dir = args.O output_dir = args.O
if (not os.path.exists(output_dir)):
os.mkdir(output_dir)
if args.m is not None: if args.m is not None:
checkpoint = torch.load(args.m, weights_only=True) checkpoint = torch.load(args.m, weights_only=True)