Spaces:
Sleeping
Sleeping
point_conf = 0.3 | |
feedb_conf = 0.3 | |
color_conf = 0.3 | |
ja_point_conf = 0.3 | |
ja_feedb_conf = 0.3 | |
ja_color_conf = 0.3 | |
emul_conf = 2 | |
rri_conf = 0.2*4 | |
op_conf = 0.5 * 3 # hidden, normal, expert | |
form_conf = 0.2*8 | |
scaf_conf = 0.2*4 | |
configurations = point_conf + feedb_conf + color_conf + ja_point_conf + ja_feedb_conf + ja_color_conf + emul_conf + rri_conf + op_conf + form_conf + scaf_conf | |
# configurations = 1 | |
# | |
configurations = 0.3 + 0.3 + 0.3 + 8*0.3 | |
# configurations = 3*0.2 + 0.5 + 0.04*2 + 0.5*2 | |
# num_of_trains = 3 + 3 + 2 + 4 + 3 + 8 + 4 | |
# print("num_of_trains:", num_of_trains) | |
configurations = 0.01 * 6 | |
print(f"Number of trains: {configurations}") | |
frames = 100_000_000 | |
# frames = 75_000_000 | |
# frames = 50_000_000 | |
seeds = 8 | |
# seeds = 4 | |
print(f"Number of seeds: {seeds}") | |
# ## one GPU | |
# fps = 300 | |
fps = 580 # ssh jz | |
# fps = 500 # ssh pf | |
gpus_per_seed = 1 | |
print(f"\n{gpus_per_seed} GPU") | |
seed_frames = frames | |
one_seed_time = 1_000_000 / (fps * 60 * 60) | |
print("train time (1M frames): {}h - {:d}d {:.0f}h".format( | |
one_seed_time, | |
int(one_seed_time // 24), one_seed_time % 24) | |
) | |
total_gpuh = configurations*seeds*gpus_per_seed*frames/(fps*60*60) | |
print("total gpu hours 1 gpups:", total_gpuh) | |
# ## half a GPU | |
# | |
# fps = 275 | |
# fps = 370 # ssh jz | |
# # fps = 300 # ssh pf | |
# gpus_per_seed = 0.5 | |
# | |
# print(f"\n{gpus_per_seed} GPU") | |
# one_seed = frames/(fps*60*60) | |
# print("train time: {}h - {:d}d {:.0f}h".format(one_seed, int(one_seed // 24), one_seed % 24)) | |
# | |
# total_gpuh = configurations*seeds*gpus_per_seed*frames/(fps*60*60) | |
# print("total gpu hours 0.5 gpups:", total_gpuh) | |
# | |
# # ## 1/3 of a GPU | |
# fps = 250 # ssh jz 1/3 | |
# # fps = 250 # ssh 1/3 pf | |
# | |
# gpus_per_seed = 0.33 | |
# print(f"\n{gpus_per_seed} GPU") | |
# | |
# one_seed = frames/(fps*60*60) | |
# print("train time: {}h - {:d}d {:.0f}h".format(one_seed, int(one_seed // 24), one_seed % 24)) | |
# | |
# total_gpuh = configurations*seeds*gpus_per_seed*frames/(fps*60*60) | |
# print("total gpu hours 0.33 gpups:", total_gpuh) | |
# | |
# | |
# # ## 1/4 of gpu | |
# # fps = 190 # ssh 1/4 pf | |
# # | |
# # gpus_per_seed = 0.25 | |
# # print(f"\n{gpus_per_seed} GPU") | |
# # | |
# # one_seed = frames/(fps*60*60) | |
# # print("train time: {}h - {:d}d {:.0f}h".format(one_seed, int(one_seed // 24), one_seed % 24)) | |
# # | |
# # total_gpuh = configurations*seeds*gpus_per_seed*frames/(fps*60*60) | |
# # print("total gpu hours 0.25 gpups:", total_gpuh) | |