mirror of
https://github.com/wassname/DeepRL.git
synced 2026-09-10 11:40:58 +08:00
Fix import
This commit is contained in:
@@ -7,6 +7,7 @@
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
import numpy as np
|
||||
|
||||
class BaseNet:
|
||||
def set_gpu(self, gpu):
|
||||
|
||||
Reference in New Issue
Block a user