123
parent
a4af74a7cb
commit
1dab8ae3e1
3
app.py
3
app.py
|
@ -14,7 +14,7 @@ from multiprocessing import Process, Event
|
|||
|
||||
app = Flask(__name__)
|
||||
|
||||
device_count=0
|
||||
device_count = torch.cuda.device_count()
|
||||
workers = {}
|
||||
|
||||
@app.route('/start/<id>')
|
||||
|
@ -191,7 +191,6 @@ def post_request(url, data):
|
|||
# break
|
||||
|
||||
if __name__ == '__main__':
|
||||
device_count = torch.cuda.device_count()
|
||||
# cProfile.run('test_function()', 'test_function.profile')
|
||||
# p = pstats.Stats('test_function.profile')
|
||||
# p.sort_stats('cumulative').print_stats(100) # Top 10 by cumulative time
|
||||
|
|
Loading…
Reference in New Issue