(source, i, args, seq_len=None, evaluation=False)
| 23 | |
| 24 | |
| 25 | def get_batch(source, i, args, seq_len=None, evaluation=False): |
| 26 | seq_len = min(seq_len if seq_len else args.bptt, len(source) - 1 - i) |
| 27 | data = Variable(source[i:i+seq_len], volatile=evaluation) |
| 28 | target = Variable(source[i+1:i+1+seq_len]) |
| 29 | return data, target |
| 30 | |
| 31 | |
| 32 | def create_exp_dir(path, scripts_to_save=None): |