robossembler/webstudio
7
0
Fork 0
Code Issues 8 Pull requests Projects Releases Packages Wiki Activity Actions

progress

Browse source
This commit is contained in:
IDONTSUDO 2024-05-02 12:40:20 +03:00
parent f0ed478b36
commit 8bc943de2c
16 changed files with 252 additions and 134 deletions
Download patch file Download diff file Expand all files Collapse all files

2
web_p/education.py
View file

@ -11,6 +11,7 @@ parser.add_argument("--datasetName", required=True, help="String with dataset na
parser.add_argument("--outpath", default="weights", help="Output path for weights")
parser.add_argument("--type", default="ObjectDetection", help="Type of implementation")
parser.add_argument("--epoch", default=3, help="How many training epochs")
parser.add_argument('--pretrain', help="Use pretraining")
args = parser.parse_args()
@ -28,5 +29,4 @@ def copy_and_move_folder(src, dst, folder):
source_folder = os.path.dirname(os.path.abspath(__file__)) + "/education"
copy_and_move_folder(source_folder, args.path, args.name)