(self,
image_size=224,
memory_size=8,
predict_size=24,
temporal_depth=8,
heads=8,
token_dim=384,
channels=3,
device='cuda:0')
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected