Move and cleanup copy tasks
This commit is contained in:
parent
978cdbb60e
commit
059b975a22
0
tasks/__init__.py
Normal file
0
tasks/__init__.py
Normal file
@ -20,7 +20,7 @@ import torch.optim as optim
|
||||
|
||||
from torch.nn.utils import clip_grad_norm
|
||||
|
||||
from dnc import DNC
|
||||
from dnc.dnc import DNC
|
||||
|
||||
parser = argparse.ArgumentParser(description='PyTorch Differentiable Neural Computer')
|
||||
parser.add_argument('-input_size', type=int, default= 6, help='dimension of input feature')
|
||||
@ -101,13 +101,6 @@ if __name__ == '__main__':
|
||||
mem_size = args.mem_size
|
||||
read_heads = args.read_heads
|
||||
|
||||
|
||||
# options, _ = getopt.getopt(sys.argv[1:], '', ['iterations='])
|
||||
|
||||
# for opt in options:
|
||||
# if opt[0] == '-iterations':
|
||||
# iterations = int(opt[1])
|
||||
|
||||
rnn = DNC(
|
||||
input_size=args.input_size,
|
||||
hidden_size=args.nhid,
|
Loading…
Reference in New Issue
Block a user