import torch from enformer_pytorch import Enformer, seq_indices_to_one_hot model = Enformer.from_hparams( dim = 1536, depth = 11, heads = 8, output_heads = dict(human = 5313, mouse = 1643), target_length = 896, ) seq = torch.randint(0, 5, (1, 196_608)) one_hot = seq_in...
import torch from enformer_pytorch import Enformer, seq_indices_to_one_hot model = Enformer.from_hparams( dim = 1536, depth = 11, heads = 8, output_heads = dict(human = 5313, mouse = 1643), target_length = 200, ).cuda() seq = torch.randint(0, 5, (196_608 // 2,)).cuda(...