SocialAISchool / gpuh.py
grg's picture
Cleaned old git history
be5548b
raw
history blame
2.36 kB
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)