File warplayer.py

model/warplayer.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	import torch
2	import torch.nn as nn
3
4	device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

nothing calls this directly

deviceMethod · 0.45

no test coverage detected