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