mirrors
/
yolov7
mirror of https://github.com/WongKinYiu/yolov7.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

claer ml debug

pull/2109/head
hanoch 2024-10-20 16:51:29 +03:00
parent 60ef66ce19
commit 102d731395
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -56,7 +56,7 @@ task = Task.init(
task_name="train yolov7 with dummy test"
)
task.set_base_docker(docker_image="nvcr.io/nvidia/pytorch:24.09-py3")
task.set_base_docker(docker_image="nvcr.io/nvidia/pytorch:24.09-py3", docker_arguments="--shm-size 8G")
gradient_clip_value = 100.0
opt_gradient_clipping = True
@ -779,6 +779,8 @@ if __name__ == '__main__':
config_file = task.connect_configuration(opt.hyp)
with open(config_file) as f:
hyp = yaml.load(f, Loader=yaml.SafeLoader) # data dict
print("", 100 * '==')
print('Hyperparameters:', hyp)
else:
# Hyperparameters
with open(opt.hyp) as f: