1 回答

TA貢獻(xiàn)1752條經(jīng)驗 獲得超4個贊
替換了這個:
%%writefile -a submission.py
model = Net()
model = model.float()
model.load_state_dict(state_dict)
model = model.to('cpu')
model = model.eval()
obs = tensor(obs['board']).reshape(1, 1, config.rows, config.columns).float()
obs = obs / 2
action = model(obs)
return int(action)
這樣:
submission_ending = ''' model = Net()
model = model.float()
model.load_state_dict(state_dict)
model = model.to('cpu')
model = model.eval()
obs = tensor(obs['board']).reshape(1, 1, config.rows, config.columns).float()
obs = obs / 2
action = model(obs)
return int(action)'''
with open('submission.py', mode='a') as file:
file.write(submission_ending)
添加回答
舉報