GPU見積もり

https://docs.google.com/spreadsheets/d/1gTWMYWcgElyGq3KK7z74ojoBP0bvwwW0PryM0NtfY2w/edit?gid=0#gid=0

現状をまとめると実装は大体終わっていて、個々のモジュールもテストした

全部通しでテストは出来ていない

ここを小規模実験でやることになる

viclip…vitだった

https://github.com/OpenGVLab/InternVideo/tree/main/Data/InternVid

Mamba2で実装できた

collect_data.py 280 torch.save(self.agent..state_dict(), f) 287 291 payload = torch.load(f, map_location=torch.device('cuda' if torch.cuda.is_available() else 'cpu'))

train.py

374

torch.save(self.agent.state_dict(), f)